[RTDETR]fix mosaic<1.0 (#3518)

2 years ago · 8c0a7440df
parent 31b46bf2b4
commit 8c0a7440df
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']