From 11f22c1ac90f23831fb6c98e9336b70983b76dfb Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Wed, 18 Apr 2018 14:18:18 +0200 Subject: k --- data/recursive_dataset.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'data') diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py index 5cb0969..7d1db84 100644 --- a/data/recursive_dataset.py +++ b/data/recursive_dataset.py @@ -11,7 +11,7 @@ class NextImage(): def append(self, path): self.next_image = path print("APPEND " + self.id + ' ' + self.next_image) - def get(self): + def get(self, index): print("GET " + self.id + ' ' + self.next_image) return self.next_image @@ -34,7 +34,7 @@ class RecursiveDataset(BaseDataset): self.transform = get_transform(opt) def __getitem__(self, index): - A_path = im.get() + A_path = im.get(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