diff options
Diffstat (limited to 'data/recursive_dataset.py')
| -rw-r--r-- | data/recursive_dataset.py | 17 |
1 files changed, 13 insertions, 4 deletions
diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py index 82b1ae4..75b2c04 100644 --- a/data/recursive_dataset.py +++ b/data/recursive_dataset.py @@ -3,14 +3,22 @@ from data.base_dataset import BaseDataset, get_transform from data.image_folder import make_dataset from PIL import Image -next_image = "lol" +class NextImage(): + def __init__(self): + self.next_image = "lol" + def append(self, path): + self.next_image = path + def get(self): + return self.next_image + +im = NextImage() class RecursiveDataset(BaseDataset): def initialize(self, opt): print('initialize') self.opt = opt self.root = opt.dataroot - next_image = opt.dataroot + 'frame_0000.png' + im.append(opt.dataroot + 'frame_0000.png') self.dir_A = os.path.join(opt.dataroot) self.A_paths = make_dataset(self.dir_A) self.A_paths = sorted(self.A_paths) @@ -18,7 +26,7 @@ class RecursiveDataset(BaseDataset): self.transform = get_transform(opt) def __getitem__(self, index): - A_path = next_image + A_path = im.get() print('next image: ' + A_path) A_img = Image.open(A_path).convert('RGB') A = self.transform(A_img) @@ -38,7 +46,8 @@ class RecursiveDataset(BaseDataset): def append(self, path): print('append', path) - next_image = path + # next_image = path + im.append(path) def name(self): return 'RecursiveImageDataset' |