Skip to content

Commit f944e91

Browse files
glenn-jocherClay Januhowski
authored andcommitted
Update dataloaders.py remove float64 shapes (ultralytics#8865)
May help ultralytics#8862
1 parent b84d0d2 commit f944e91

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

utils/dataloaders.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -478,7 +478,7 @@ def __init__(self,
478478
[cache.pop(k) for k in ('hash', 'version', 'msgs')] # remove items
479479
labels, shapes, self.segments = zip(*cache.values())
480480
self.labels = list(labels)
481-
self.shapes = np.array(shapes, dtype=np.float64)
481+
self.shapes = np.array(shapes)
482482
self.im_files = list(cache.keys()) # update
483483
self.label_files = img2label_paths(cache.keys()) # update
484484
n = len(shapes) # number of images

0 commit comments

Comments
 (0)