From a2637bbedf7d93bce0e013d57c8d665960e668d3 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Wed, 18 Apr 2018 14:21:04 +0200
Subject: k

---
 data/recursive_dataset.py | 7 ++++---
 1 file 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'
-- 
cgit v1.2.3-70-g09d2