Update dataloaders.py remove `float64` shapes (#8865)

May help https://github.com/ultralytics/yolov5/issues/8862
2022-08-04 17:06:08 +02:00 · 2022-08-04 17:06:08 +02:00 · 84e7748564
parent 628c05ca6f
commit 84e7748564
1 changed files with 1 additions and 1 deletions
--- a/utils/dataloaders.py
+++ b/utils/dataloaders.py
@ -478,7 +478,7 @@ class LoadImagesAndLabels(Dataset):
        [cache.pop(k) for k in ('hash', 'version', 'msgs')]  # remove items
        labels, shapes, self.segments = zip(*cache.values())
        self.labels = list(labels)
-        self.shapes = np.array(shapes, dtype=np.float64)
+        self.shapes = np.array(shapes)
        self.im_files = list(cache.keys())  # update
        self.label_files = img2label_paths(cache.keys())  # update
        n = len(shapes)  # number of images