We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
float64
1 parent b84d0d2 commit f944e91Copy full SHA for f944e91
utils/dataloaders.py
@@ -478,7 +478,7 @@ def __init__(self,
478
[cache.pop(k) for k in ('hash', 'version', 'msgs')] # remove items
479
labels, shapes, self.segments = zip(*cache.values())
480
self.labels = list(labels)
481
- self.shapes = np.array(shapes, dtype=np.float64)
+ self.shapes = np.array(shapes)
482
self.im_files = list(cache.keys()) # update
483
self.label_files = img2label_paths(cache.keys()) # update
484
n = len(shapes) # number of images
0 commit comments