more live code clean up and options

1 files changed, 1 insertions, 9 deletions

diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py
index f56b871..0daa169 100644
--- a/data/recursive_dataset.py
+++ b/data/recursive_dataset.py
@@ -14,15 +14,12 @@ class RecursiveDataset(BaseDataset):
         self.dir_A = os.path.join(opt.dataroot)
         self.A_paths = make_dataset(self.dir_A)
         self.A_paths = sorted(self.A_paths)
-
-        self.stuff = [opt.dataroot + 'frame_00000.png']
-
         self.transform = get_transform(opt)
 
     def __getitem__(self, index):
         A_path = os.path.join(self.opt.render_dir, "frame_{:05d}.png".format(index))
         while not os.path.exists(A_path):
-            time.sleep(0.01)
+            time.sleep(self.opt.poll_delay)
         A_img = Image.open(A_path).convert('RGB')
         A = self.transform(A_img)
         if self.opt.which_direction == 'BtoA':
@@ -39,10 +36,5 @@ class RecursiveDataset(BaseDataset):
     def __len__(self):
         return 10000000
 
-    def append(self, path):
-        print('append', path)
-        # next_image = path
-        self.stuff.append(path)
-
     def name(self):
         return 'RecursiveImageDataset'