summaryrefslogtreecommitdiff
path: root/data/recursive_dataset.py
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2018-06-17 09:25:42 +0200
committerJules Laplace <julescarbon@gmail.com>2018-06-17 09:25:42 +0200
commit897d1e830d730240eac636f0c8ceb31fc5dd8bd7 (patch)
treec6c6bad312c86bfd54fe2ea35f561a01cf07bba9 /data/recursive_dataset.py
parent48faae06f6f7cf749e328f07a7613aa74fa1a3df (diff)
args
Diffstat (limited to 'data/recursive_dataset.py')
-rw-r--r--data/recursive_dataset.py2
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):