From d3e77781517fe5a28bd929b947278c832f95c036 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sat, 7 Nov 2020 02:27:31 +0100
Subject: [PATCH] imap bug fix
---
utils/datasets.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/utils/datasets.py b/utils/datasets.py
index 45b950ec8..e069568ec 100755
--- a/utils/datasets.py
+++ b/utils/datasets.py
@@ -477,7 +477,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing
if cache_images:
gb = 0 # Gigabytes of cached images
self.img_hw0, self.img_hw = [None] * n, [None] * n
- results = ThreadPool(8).imap_unordered(lambda x: load_image(*x), zip(repeat(self), range(n))) # 8 threads
+ results = ThreadPool(8).imap(lambda x: load_image(*x), zip(repeat(self), range(n))) # 8 threads
pbar = tqdm(enumerate(results), total=n)
for i, x in pbar:
self.imgs[i], self.img_hw0[i], self.img_hw[i] = x # img, hw_original, hw_resized = load_image(self, i)