From e3079a7496607d988aca02984fa60e3c607d8ee6 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Thu, 4 Aug 2022 17:01:07 +0200
Subject: [PATCH] Update dataloaders.py remove `float64` shapes

May help https://github.com/ultralytics/yolov5/issues/8862
---
 utils/dataloaders.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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