Reference for ultralytics/models/yolo/segment/train.py
Note
This file is available at https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/segment/train.py. If you spot a problem please help fix it by contributing a Pull Request 🛠️. Thank you 🙏!
ultralytics.models.yolo.segment.train.SegmentationTrainer
SegmentationTrainer(
cfg=DEFAULT_CFG, overrides: Optional[Dict] = None, _callbacks=None
)
Bases: DetectionTrainer
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:
Name | Type | Description |
---|---|---|
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()
This initializes a trainer for segmentation tasks, extending the detection trainer with segmentation-specific functionality. It sets the task to 'segment' and prepares the trainer for training segmentation models.
Parameters:
Name | Type | Description | Default |
---|---|---|---|
cfg
|
dict
|
Configuration dictionary with default training settings. |
DEFAULT_CFG
|
overrides
|
dict
|
Dictionary of parameter overrides for the default configuration. |
None
|
_callbacks
|
list
|
List of callback functions to be executed during training. |
None
|
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()
Source code in ultralytics/models/yolo/segment/train.py
30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
|
get_model
get_model(
cfg: Optional[Union[Dict, str]] = None,
weights: Optional[Union[str, Path]] = None,
verbose: bool = True,
)
Initialize and return a SegmentationModel with specified configuration and weights.
Parameters:
Name | Type | Description | Default |
---|---|---|---|
cfg
|
dict | str
|
Model configuration. Can be a dictionary, a path to a YAML file, or None. |
None
|
weights
|
str | Path
|
Path to pretrained weights file. |
None
|
verbose
|
bool
|
Whether to display model information during initialization. |
True
|
Returns:
Type | Description |
---|---|
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)
Source code in ultralytics/models/yolo/segment/train.py
53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 |
|
get_validator
get_validator()
Return an instance of SegmentationValidator for validation of YOLO model.
Source code in ultralytics/models/yolo/segment/train.py
78 79 80 81 82 83 |
|
plot_metrics
plot_metrics()
Plot training/validation metrics.
Source code in ultralytics/models/yolo/segment/train.py
85 86 87 |
|