From f49d7f6397d9e7c3e256ce01cc2dd33b84714987 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Wed, 18 Apr 2018 19:34:55 +0200
Subject: k

---
 data/recursive_dataset.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'data')

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)
-- 
cgit v1.2.3-70-g09d2