[RTDETR]fix mosaic<1.0 (#3518)

2023-07-04 16:09:17 +08:00 · 2023-07-04 16:09:17 +08:00 · 8c0a7440df
commit 8c0a7440df
parent 31b46bf2b4
1 changed files with 1 additions and 1 deletions
--- a/ultralytics/yolo/data/augment.py
+++ b/ultralytics/yolo/data/augment.py
@ -427,7 +427,7 @@ class RandomPerspective:
        """
        if self.pre_transform and 'mosaic_border' not in labels:
            labels = self.pre_transform(labels)
-            labels.pop('ratio_pad')  # do not need ratio pad
+        labels.pop('ratio_pad', None)  # do not need ratio pad

        img = labels['img']
        cls = labels['cls']