YOLO Vision 2026:

Reference for ultralytics/models/yolo/yoloe/train_seg.py#

Improvements

This page is sourced from https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/yoloe/train_seg.py. Have an improvement or example to add? Open a Pull Request — thank you! 🙏


Summary

Class ultralytics.models.yolo.yoloe.train_seg.YOLOESegTrainer#

YOLOESegTrainer()

Bases: YOLOETrainer, SegmentationTrainer

Trainer class for YOLOE segmentation models.

GitHubultralytics/models/yolo/yoloe/train_seg.py
class YOLOESegTrainer(YOLOETrainer, SegmentationTrainer):
    """Trainer class for YOLOE segmentation models."""




Class ultralytics.models.yolo.yoloe.train_seg.YOLOEPESegTrainer#

YOLOEPESegTrainer()

Bases: SegmentationTrainer

Fine-tune YOLOE segmentation models in linear probing way.

GitHubultralytics/models/yolo/yoloe/train_seg.py
class YOLOEPESegTrainer(SegmentationTrainer):
    """Fine-tune YOLOE segmentation models in linear probing way."""

    get_model = YOLOEPETrainer.get_model  # shared linear-probing builder; SegmentationTrainer stays the sole base




Class ultralytics.models.yolo.yoloe.train_seg.YOLOESegTrainerFromScratch#

YOLOESegTrainerFromScratch()

Bases: YOLOETrainerFromScratch, YOLOESegTrainer

Trainer for YOLOE segmentation models trained from scratch without pretrained weights.

GitHubultralytics/models/yolo/yoloe/train_seg.py
class YOLOESegTrainerFromScratch(YOLOETrainerFromScratch, YOLOESegTrainer):
    """Trainer for YOLOE segmentation models trained from scratch without pretrained weights."""




Class ultralytics.models.yolo.yoloe.train_seg.YOLOESegVPTrainer#

YOLOESegVPTrainer()

Bases: YOLOEVPTrainer, YOLOESegTrainerFromScratch

Trainer for YOLOE segmentation models with Vision Prompt (VP) capabilities.

GitHubultralytics/models/yolo/yoloe/train_seg.py
class YOLOESegVPTrainer(YOLOEVPTrainer, YOLOESegTrainerFromScratch):
    """Trainer for YOLOE segmentation models with Vision Prompt (VP) capabilities."""



Contributors
GLglenn-jocher3RIRizwanMunawar1