.h ddlmZddlmZddlmZddlmZddlmZddl m Z m Z GddejjZy ) ) annotations)copy)Path)yolo)SegmentationModel) DEFAULT_CFGRANKc<eZdZdZeddfdfd ZdddZdZxZS) SegmentationTrainera A class extending the DetectionTrainer class for training based on a segmentation model. This trainer specializes in handling segmentation tasks, extending the detection trainer with segmentation-specific functionality including model initialization, validation, and visualization. Attributes: loss_names (tuple[str]): Names of the loss components used during training. Examples: >>> from ultralytics.models.yolo.segment import SegmentationTrainer >>> args = dict(model="yolo11n-seg.pt", data="coco8-seg.yaml", epochs=3) >>> trainer = SegmentationTrainer(overrides=args) >>> trainer.train() Nc:|i}d|d<t||||y)ag Initialize a SegmentationTrainer object. Args: cfg (dict): Configuration dictionary with default training settings. overrides (dict, optional): Dictionary of parameter overrides for the default configuration. _callbacks (list, optional): List of callback functions to be executed during training. Nsegmenttask)super__init__)selfcfg overrides _callbacks __class__s c/var/www/html/ai-service/venv/lib/python3.12/site-packages/ultralytics/models/yolo/segment/train.pyrzSegmentationTrainer.__init__s+  I% & i4ct||jd|jd|xr tdk(}|r|j||S)a Initialize and return a SegmentationModel with specified configuration and weights. Args: cfg (dict | str, optional): Model configuration. Can be a dictionary, a path to a YAML file, or None. weights (str | Path, optional): Path to pretrained weights file. verbose (bool): Whether to display model information during initialization. Returns: (SegmentationModel): Initialized segmentation model with loaded weights if specified. Examples: >>> trainer = SegmentationTrainer() >>> model = trainer.get_model(cfg="yolo11n-seg.yaml") >>> model = trainer.get_model(weights="yolo11n-seg.pt", verbose=False) ncchannels)rchverbose)rdatar load)rrweightsrmodels r get_modelzSegmentationTrainer.get_model,sF""#$))D/dii >S]d]simqsist  JJw  rcd|_tjj|j|j t |j|jS)zIReturn an instance of SegmentationValidator for validation of YOLO model.)box_lossseg_losscls_lossdfl_loss)save_dirargsr) loss_namesrr SegmentationValidator test_loaderr(rr) callbacks)rs r get_validatorz!SegmentationTrainer.get_validatorCsGH||11   t}}4 ?W[WeWe2  r)rz dict | None)NNT)rzdict | str | Noner zstr | Path | Nonerbool) __name__ __module__ __qualname____doc__rrr"r. __classcell__)rs@rr r s! 'RV 5. rr N) __future__rrpathlibrultralytics.modelsrultralytics.nn.tasksrultralytics.utilsrr detectDetectionTrainerr rrr=s/##2/; $++66; r