diff options
Diffstat (limited to 'data/recursive_dataset.py')
| -rw-r--r-- | data/recursive_dataset.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py index ca74ec3..e8d6c70 100644 --- a/data/recursive_dataset.py +++ b/data/recursive_dataset.py @@ -16,7 +16,7 @@ class RecursiveDataset(BaseDataset): def __getitem__(self, index): ### input A (label maps) - A_path = os.path.join(self.opt.dataroot, "frame_{:05d}.png".format(index)) + A_path = os.path.join(self.opt.render_dir, "frame_{:05d}.png".format(index)) if not os.path.exists(A_path): print("{} doesn't exist, waiting for it".format(A_path)) while not os.path.exists(A_path): |