h SSKJr SSKJr SSKJr SSKJr SSKJr SSK J r J r "SS\RR5rg ) ) annotations)copy)Path)yolo)SegmentationModel) DEFAULT_CFGRANKcN^\rSrSrSr\SS4SU4SjjjrS S SjjrSrSr U=r $) SegmentationTrainer ag 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:>Uc0nSUS'[TU]XU5 g)a7 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 _/home/james-whalen/.local/lib/python3.13/site-packages/ultralytics/models/yolo/segment/train.pyrSegmentationTrainer.__init__s)  I% & 4c[XRSURSU=(a [S:HS9nU(aURU5 U$)aw 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_modelSegmentationTrainer.get_model,sD""#))D/dii >S]d]simqsist  JJw  rcSUl[RRURUR [ UR5URS9$)zIReturn an instance of SegmentationValidator for validation of YOLO model.)box_lossseg_losscls_lossdfl_loss)save_dirargsr) loss_namesrrSegmentationValidator test_loaderr+rr, callbacks)rs r get_validator!SegmentationTrainer.get_validatorCsGH||11   t}}4 ?W[WeWe2  r)r-)rz dict | None)NNT)rzdict | str | Noner"zstr | Path | Nonerbool) __name__ __module__ __qualname____firstlineno____doc__rrr$r1__static_attributes__ __classcell__)rs@rr r s+ 'RV 5 5.  rr N) __future__rrpathlibrultralytics.modelsrultralytics.nn.tasksrultralytics.utilsrr detectDetectionTrainerr rrrCs/##2/; $++66; r