diff options
| -rw-r--r-- | data/recursive_dataset.py | 19 | ||||
| -rw-r--r-- | test.py | 2 |
2 files changed, 2 insertions, 19 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) @@ -22,7 +22,7 @@ if __name__ == '__main__': render_dir = opt.results_dir + "/" + opt.name + "/exp:" + opt.experiment mkdirs(render_dir) - copyfile(opt.start_img, render_dir) + copyfile(opt.start_img, render_dir + "/frame_00000.png") data_loader = CreateRecursiveDataLoader(opt) dataset = data_loader.load_data() |