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

May help ultralytics#8862
ctjanuhowski · Sep 8, 2022 · f944e91 · f944e91
1 parent b84d0d2
commit f944e91
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