Fix numpy FP64 bug in augmentation (#2841)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 years ago · ef1eff8830
parent 527a97759b
commit ef1eff8830
1 changed files with 1 additions and 1 deletions
--- a/ultralytics/yolo/data/augment.py
+++ b/ultralytics/yolo/data/augment.py
@ -683,7 +683,7 @@ class Albumentations:
                if len(new['class_labels']) > 0:  # skip update if no bbox in new im
                    labels['img'] = new['image']
                    labels['cls'] = np.array(new['class_labels'])
-                    bboxes = np.array(new['bboxes'])
+                    bboxes = np.array(new['bboxes'], dtype=np.float32)
            labels['instances'].update(bboxes=bboxes)
        return labels