Reference for ultralytics/models/yolo/segment/predict.py
Note
This file is available at https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/segment/predict.py. If you spot a problem please help fix it by contributing a Pull Request 🛠️. Thank you 🙏!
ultralytics.models.yolo.segment.predict.SegmentationPredictor
Bases: DetectionPredictor
A class extending the DetectionPredictor class for prediction based on a segmentation model.
Example
Source code in ultralytics/models/yolo/segment/predict.py
postprocess
Applies non-max suppression and processes detections for each image in an input batch.
Source code in ultralytics/models/yolo/segment/predict.py
📅 Created 1 year ago
✏️ Updated 2 months ago