Unverified 提交 84e77485 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

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

上级 628c05ca
......@@ -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
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论