y

1 files changed, 1 insertions, 18 deletions

diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py
index f8a5da8..e42f1f7 100644
--- a/data/recursive_dataset.py
+++ b/data/recursive_dataset.py
@@ -6,23 +6,6 @@ from data.image_folder import make_dataset
 from PIL import Image
 import random
 
-class NextImage():
-    def __init__(self):
-        self.next_image = "lol"
-        self.id = str(random.randint(1,100))
-    def append(self, path):
-        self.next_image = path
-        print("APPEND " + self.id + ' ' + self.next_image)
-    def get(self, index):
-        print("GET " + self.id + ' ' + self.next_image)
-        return self.next_image
-
-im = NextImage()
-im.append('a')
-im.get(1)
-im.append('b')
-im.get(1)
-
 class RecursiveDataset(BaseDataset):
     def initialize(self, opt):
         print('initialize')
@@ -37,7 +20,7 @@ class RecursiveDataset(BaseDataset):
         self.transform = get_transform(opt)
 
     def __getitem__(self, index):
-        A_path = self.opt.results_dir + "/" + self.opt.name + "/exp:" + self.opt.experiment + "/frame_{:05d}.png".format(index)
+        A_path = self.opt.results_dir + self.opt.name + "/exp:" + self.opt.experiment + "/frame_{:05d}.png".format(index)
         print('next image: ' + A_path)
         while not os.path.exists(A_path):
             time.sleep(0.1)