From 3b55c1e130d0466e164a8344a9fb66e504ad8bbc Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Thu, 19 Apr 2018 02:18:38 +0200 Subject: y --- data/recursive_dataset.py | 19 +------------------ 1 file changed, 1 insertion(+), 18 deletions(-) (limited to 'data/recursive_dataset.py') 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) -- cgit v1.2.3-70-g09d2