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 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) |