Update datasets.py

ZHEQIUSHUI · Nov 27, 2023 · 84b1538 · 84b1538
1 parent 875d0c1
commit 84b1538
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/yolov6/data/datasets.py b/yolov6/data/datasets.py
@@ -141,7 +141,7 @@ def cache_images(self, num_imgs=None):
         load_imgs = ThreadPool(num_threads).imap(self.load_image, range(num_imgs))
         pbar = tqdm(enumerate(load_imgs), total=num_imgs, disable=self.rank > 0)
         for i, (x, (h0, w0), shape) in pbar:
-            self.imgs[i], self.imgs_hw0[i], self.imgs_hw = x, (h0, w0), shape
+            self.imgs[i], self.imgs_hw0[i], self.imgs_hw[i] = x, (h0, w0), shape
 
     def __del__(self):
         if self.cache_ram: