summaryrefslogtreecommitdiff
path: root/data/recursive_dataset.py
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2018-05-15 02:42:33 +0200
committerJules Laplace <julescarbon@gmail.com>2018-05-15 02:42:33 +0200
commit743c7ee58013ab6095ee545e723fa06704350f14 (patch)
tree2be98257cbe16ba67478441e5b8a48c3ea6c5855 /data/recursive_dataset.py
parent86a8bc3914d455a16027d35e7a056bcc5ce189c1 (diff)
test
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 156f161..762e391 100644
--- a/data/recursive_dataset.py
+++ b/data/recursive_dataset.py
@@ -20,7 +20,7 @@ class RecursiveDataset(BaseDataset):
self.transform = get_transform(opt)
def __getitem__(self, index):
- A_path = self.opt.results_dir + self.opt.name + "/exp:" + self.opt.experiment + "/frame_{:05d}.png".format(index)
+ A_path = self.opt.results_dir + self.opt.tag + "/frame_{:05d}.png".format(index)
while not os.path.exists(A_path):
time.sleep(0.01)
A_img = Image.open(A_path).convert('RGB')