Fi ddlmZddlmZddlmZddlmZddlmZddl m Z m Z GddejjZy ) ) annotations)copy)Path)yolo)SegmentationModel) DEFAULT_CFGRANKc<eZdZdZeddfdfd ZdddZdZxZS) SegmentationTraineraA 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="yolo26n-seg.pt", data="coco8-seg.yaml", epochs=3) >>> trainer = SegmentationTrainer(overrides=args) >>> trainer.train() Nc:|i}d|d<t||||y)adInitialize a SegmentationTrainer object. Args: cfg (dict): Configuration dictionary with default training settings. overrides (dict, optional): Dictionary of parameter overrides for the default configuration. _callbacks (dict, optional): Dictionary of callback functions to be executed during training. Nsegmenttask)super__init__)selfcfg overrides _callbacks __class__s g/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/ultralytics/models/yolo/segment/train.pyrzSegmentationTrainer.__init__s+  I% & i4ct||jd|jd|xr tdk(}|r|j||S)aInitialize 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="yolo26n-seg.yaml") >>> model = trainer.get_model(weights="yolo26n-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_losssem_loss)save_dirargsr) loss_namesrr SegmentationValidator test_loaderr)rr* callbacks)rs r get_validatorz!SegmentationTrainer.get_validator@sGT||11   t}}4 ?W[WeWe2  r)r dict | Nonerr0)NNT)rzdict | str | Noner zstr | Path | Nonerbool) __name__ __module__ __qualname____doc__rrr"r/ __classcell__)rs@rr r s! 'ae 5, rr N) __future__rrpathlibrultralytics.modelsrultralytics.nn.tasksrultralytics.utilsrr detectDetectionTrainerr rrr?s/##2/8 $++668 r