From 8c3efb2040c603afebd8c86b4387e334d07c7c35 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Wed, 18 Apr 2018 14:06:26 +0200
Subject: k

---
 data/recursive_dataset.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

(limited to 'data')

diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py
index 2560c3b..40ac452 100644
--- a/data/recursive_dataset.py
+++ b/data/recursive_dataset.py
@@ -3,12 +3,13 @@ from data.base_dataset import BaseDataset, get_transform
 from data.image_folder import make_dataset
 from PIL import Image
 
+next_image = "lol"
 
 class RecursiveDataset(BaseDataset):
     def initialize(self, opt):
         self.opt = opt
         self.root = opt.dataroot
-        self.next_image = opt.dataroot + 'frame_0000.png'
+        next_image = opt.dataroot + 'frame_0000.png'
         self.dir_A = os.path.join(opt.dataroot)
         self.A_paths = make_dataset(self.dir_A)
         self.A_paths = sorted(self.A_paths)
@@ -16,7 +17,7 @@ class RecursiveDataset(BaseDataset):
         self.transform = get_transform(opt)
 
     def __getitem__(self, index):
-        A_path = self.next_image
+        A_path = next_image
         print('next image: ' + A_path)
         A_img = Image.open(A_path).convert('RGB')
         A = self.transform(A_img)
@@ -36,7 +37,7 @@ class RecursiveDataset(BaseDataset):
 
     def append(self, path):
         print('append', path)
-        self.next_image = path
+        next_image = path
 
     def name(self):
         return 'RecursiveImageDataset'
-- 
cgit v1.2.3-70-g09d2