summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2018-04-18 19:34:55 +0200
committerJules Laplace <julescarbon@gmail.com>2018-04-18 19:34:55 +0200
commitf49d7f6397d9e7c3e256ce01cc2dd33b84714987 (patch)
treeb1e78dea92414094698e4af26dbc24adb6820d1e
parent919ba03dae6ca25efc2f0bafc5a027efc0fe18e1 (diff)
k
-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 176aee3..b4135dc 100644
--- a/data/recursive_dataset.py
+++ b/data/recursive_dataset.py
@@ -36,7 +36,7 @@ class RecursiveDataset(BaseDataset):
self.transform = get_transform(opt)
def __getitem__(self, index):
- A_path = self.stuff[index]
+ A_path = "recursive/frame_{:04d}.png".format(index)
print('next image: ' + A_path)
A_img = Image.open(A_path).convert('RGB')
A = self.transform(A_img)