diff options
| -rw-r--r-- | data/recursive_dataset.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py index 1ec5e19..16f30fd 100644 --- a/data/recursive_dataset.py +++ b/data/recursive_dataset.py @@ -19,18 +19,18 @@ class RecursiveDataset(BaseDataset): def __getitem__(self, index): A_path = os.path.join(self.opt.render_dir, "frame_{:05d}.png".format(index)) - print(A_path) + # print(A_path) if not os.path.exists(A_path): A_path2 = os.path.join(self.opt.render_dir, "frame_{:05d}.png".format(index+1)) if os.path.exists(A_path2): - print('oops: {}'.format(A_path2)) + # print('oops: {}'.format(A_path2)) A_path = A_path2 else: - print('not there yet... {}'.format(A_path)) + # print('not there yet... {}'.format(A_path)) while not os.path.exists(A_path): - print('sleeping for {}'.format(self.opt.poll_delay)) + # print('sleeping for {}'.format(self.opt.poll_delay)) time.sleep(self.opt.poll_delay) - print("got {}".format(A_path)) + # print("got {}".format(A_path)) A_img = Image.open(A_path).convert('RGB') A = self.transform(A_img) if self.opt.which_direction == 'BtoA': |