diff options
Diffstat (limited to 'data')
| -rw-r--r-- | data/recursive_dataset.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py index b6ce65d..1ec5e19 100644 --- a/data/recursive_dataset.py +++ b/data/recursive_dataset.py @@ -4,7 +4,8 @@ from data.base_dataset import BaseDataset, get_transform from data.image_folder import make_dataset from PIL import Image import random -import gevent +# import gevent +import time class RecursiveDataset(BaseDataset): def initialize(self, opt): @@ -26,9 +27,9 @@ class RecursiveDataset(BaseDataset): A_path = A_path2 else: print('not there yet... {}'.format(A_path)) - print('sleeping for {}'.format(self.opt.poll_delay)) while not os.path.exists(A_path): - gevent.sleep(self.opt.poll_delay) + print('sleeping for {}'.format(self.opt.poll_delay)) + time.sleep(self.opt.poll_delay) print("got {}".format(A_path)) A_img = Image.open(A_path).convert('RGB') A = self.transform(A_img) |