diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-05-21 03:23:57 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-05-21 03:23:57 +0200 |
| commit | e5ec83cf2bcffee392da274fc91ea71074bf96ff (patch) | |
| tree | 8a14dd9bb9defb9306da1c44c647ebd59c5ef23c /data/recursive_dataset.py | |
| parent | 8b6d40b3e1cc2a7164145142be2c00afd99e2ba6 (diff) | |
un rato
Diffstat (limited to 'data/recursive_dataset.py')
| -rw-r--r-- | data/recursive_dataset.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py index 0daa169..d7f1e4a 100644 --- a/data/recursive_dataset.py +++ b/data/recursive_dataset.py @@ -5,6 +5,7 @@ from data.base_dataset import BaseDataset, get_transform from data.image_folder import make_dataset from PIL import Image import random +import gevent class RecursiveDataset(BaseDataset): def initialize(self, opt): @@ -19,7 +20,7 @@ class RecursiveDataset(BaseDataset): def __getitem__(self, index): A_path = os.path.join(self.opt.render_dir, "frame_{:05d}.png".format(index)) while not os.path.exists(A_path): - time.sleep(self.opt.poll_delay) + gevent.sleep(self.opt.poll_delay) A_img = Image.open(A_path).convert('RGB') A = self.transform(A_img) if self.opt.which_direction == 'BtoA': |