k

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'