diff --git a/utils/dataloaders.py b/utils/dataloaders.py index 9ccfe2545d75..71e7428d4dc1 100755 --- a/utils/dataloaders.py +++ b/utils/dataloaders.py @@ -478,7 +478,7 @@ def __init__(self, [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