Glenn Jocher
4 years ago
1 changed files with 1 additions and 1 deletions
+ 1
- 1
utils/datasets.py
View File
| if cache_images: | if cache_images: | ||||
| gb = 0 # Gigabytes of cached images | gb = 0 # Gigabytes of cached images | ||||
| self.img_hw0, self.img_hw = [None] * n, [None] * n | 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) | pbar = tqdm(enumerate(results), total=n) | ||||
| for i, x in pbar: | 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) | self.imgs[i], self.img_hw0[i], self.img_hw[i] = x # img, hw_original, hw_resized = load_image(self, i) |
Loading…