ultralytics 8.0.153 YOLO Tasks Cleanup (#4314)

2023-08-12 02:30:57 +02:00
parent 39395aedc8
commit 822608986c
22 changed files with 87 additions and 55 deletions
--- a/ultralytics/models/fastsam/val.py
+++ b/ultralytics/models/fastsam/val.py
@ -44,7 +44,7 @@ class FastSAMValidator(DetectionValidator):
                                         'R', 'mAP50', 'mAP50-95)')

    def postprocess(self, preds):
-        """Postprocesses YOLO predictions and returns output detections with proto."""
+        """Post-processes YOLO predictions and returns output detections with proto."""
        p = ops.non_max_suppression(preds[0],
                                    self.args.conf,
                                    self.args.iou,