diff options
| -rw-r--r-- | data/recursive_dataset.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py index 10513a6..176aee3 100644 --- a/data/recursive_dataset.py +++ b/data/recursive_dataset.py @@ -31,10 +31,12 @@ class RecursiveDataset(BaseDataset): self.A_paths = make_dataset(self.dir_A) self.A_paths = sorted(self.A_paths) + self.stuff = [opt.dataroot + 'frame_0000.png'] + self.transform = get_transform(opt) def __getitem__(self, index): - A_path = im.get(index) + A_path = self.stuff[index] print('next image: ' + A_path) A_img = Image.open(A_path).convert('RGB') A = self.transform(A_img) @@ -55,8 +57,7 @@ class RecursiveDataset(BaseDataset): def append(self, path): print('append', path) # next_image = path - im.append(path) - im.get(path) + self.stuff.append(path) def name(self): return 'RecursiveImageDataset' |