Add dota8.yaml and O tests (#7394)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-01-09 02:54:09 +08:00 · 2024-01-09 02:54:09 +08:00 · a6a2c256d4
commit a6a2c256d4
parent d0562d7a2f
13 changed files with 176 additions and 16 deletions
--- a/ultralytics/models/yolo/segment/train.py
+++ b/ultralytics/models/yolo/segment/train.py
@ -51,7 +51,7 @@ class SegmentationTrainer(yolo.detect.DetectionTrainer):
                    batch['batch_idx'],
                    batch['cls'].squeeze(-1),
                    batch['bboxes'],
-                    batch['masks'],
+                    masks=batch['masks'],
                    paths=batch['im_file'],
                    fname=self.save_dir / f'train_batch{ni}.jpg',
                    on_plot=self.on_plot)