diff --git a/utils/datasets.py b/utils/datasets.py index 45b950e..e069568 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)