Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Update dataloaders.py remove float64 shapes (#8865)

Browse Source 
May help https://github.com/ultralytics/yolov5/issues/8862
This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-08-04 17:06:08 +02:00
1 parent 628c05ca6f
commit 84e7748564
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
utils/dataloaders.py
+1 -1
View File
@@ -478,7 +478,7 @@ class LoadImagesAndLabels(Dataset):
[cache.pop(k) for k in ('hash', 'version', 'msgs')] # remove items [cache.pop(k) for k in ('hash', 'version', 'msgs')] # remove items
labels, shapes, self.segments = zip(*cache.values()) labels, shapes, self.segments = zip(*cache.values())
self.labels = list(labels) 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.im_files = list(cache.keys()) # update
self.label_files = img2label_paths(cache.keys()) # update self.label_files = img2label_paths(cache.keys()) # update
n = len(shapes) # number of images n = len(shapes) # number of images