Update dataloaders.py remove float64 shapes (#8865)

May help #8862
ultralytics · Aug 4, 2022 · 84e7748 · 84e7748
1 parent 628c05c
commit 84e7748
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/dataloaders.py b/utils/dataloaders.py
@@ -478,7 +478,7 @@ def __init__(self,
         [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