Reference for ultralytics/models/yolo/segment/train.py
Note
Full source code for this file is available at https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/segment/train.py. Help us fix any issues you see by submitting a Pull Request 🛠️. Thank you 🙏!
ultralytics.models.yolo.segment.train.SegmentationTrainer
Bases: DetectionTrainer
A class extending the DetectionTrainer class for training based on a segmentation model.
Example
Source code in ultralytics/models/yolo/segment/train.py
__init__(cfg=DEFAULT_CFG, overrides=None, _callbacks=None)
Initialize a SegmentationTrainer object with given arguments.
Source code in ultralytics/models/yolo/segment/train.py
get_model(cfg=None, weights=None, verbose=True)
Return SegmentationModel initialized with specified config and weights.
Source code in ultralytics/models/yolo/segment/train.py
get_validator()
Return an instance of SegmentationValidator for validation of YOLO model.
Source code in ultralytics/models/yolo/segment/train.py
plot_metrics()
plot_training_samples(batch, ni)
Creates a plot of training sample images with labels and box coordinates.