mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

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

May help https://github.com/ultralytics/yolov5/issues/8862
pull/8871/head
Glenn Jocher 2022-08-04 17:06:08 +02:00 committed by GitHub
parent 628c05ca6f
commit 84e7748564
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/dataloaders.py
View File

@ -478,7 +478,7 @@ class LoadImagesAndLabels(Dataset):
[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