From e5ec83cf2bcffee392da274fc91ea71074bf96ff Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Mon, 21 May 2018 03:23:57 +0200 Subject: un rato --- data/recursive_dataset.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'data') 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': -- cgit v1.2.3-70-g09d2