/j ddlmZddlmZddlmZddlmZddlmZddl m Z m Z Gddej j Zd S) ) 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() N overrides dict | None _callbacksc`|i}d|d<t|||dS)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__)selfcfgr r __class__s j/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/ultralytics/models/yolo/segment/train.pyrzSegmentationTrainer.__init__s<  I% & i44444Trdict | str | Noneweightsstr | Path | Noneverboseboolct||jd|jd|o tdk}|r|||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)rchr)rdatar load)rrrrmodels r get_modelzSegmentationTrainer.get_model*sS "#$)D/di >S]d]simqsisttt  JJw    rcd|_tj|j|jt |j|jS)zIReturn an instance of SegmentationValidator for validation of YOLO model.)box_lossseg_losscls_lossdfl_losssem_loss)save_dirargsr) loss_namesrrSegmentationValidator test_loaderr-rr. callbacks)rs r get_validatorz!SegmentationTrainer.get_validator@sBT|11  t}4 ??W[We2   r)r r rr )NNT)rrrrrr) __name__ __module__ __qualname____doc__rrr&r3 __classcell__)rs@rr r s~  'ae 5 5 5 5 5 5 5,       rr N) __future__rrpathlibrultralytics.modelsrultralytics.nn.tasksrultralytics.utilsrr detectDetectionTrainerr rrrAs#"""""######222222////////8 8 8 8 8 $+68 8 8 8 8 r