From fd3198c0c799e7943f7f27758e97670535c94979 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Sun, 2 Sep 2018 22:09:49 +0200 Subject: augment script --- data/recursive_dataset.py | 2 +- data/sequence_dataset.py | 12 ++++++------ 2 files changed, 7 insertions(+), 7 deletions(-) (limited to 'data') diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py index 40b7ebf..dda9c95 100644 --- a/data/recursive_dataset.py +++ b/data/recursive_dataset.py @@ -13,7 +13,7 @@ class RecursiveDataset(BaseDataset): ### input A (label maps) self.dataset_size = 1000000 - + def __getitem__(self, index): ### input A (label maps) A_path = os.path.join(self.opt.render_dir, "frame_{:05d}.png".format(index)) diff --git a/data/sequence_dataset.py b/data/sequence_dataset.py index 3eaa12b..c3c7d44 100644 --- a/data/sequence_dataset.py +++ b/data/sequence_dataset.py @@ -8,18 +8,18 @@ from PIL import Image class SequenceDataset(BaseDataset): def initialize(self, opt): self.opt = opt - self.root = opt.dataroot + self.root = opt.dataroot ### input A (label maps) self.dir_A = opt.dataroot self.A_paths = sorted(make_dataset(self.dir_A)) - self.dataset_size = len(self.A_paths) - - def __getitem__(self, index): + self.dataset_size = len(self.A_paths) + + def __getitem__(self, index): ### input A (label maps) - A_path = self.A_paths[index] - A = Image.open(A_path) + A_path = self.A_paths[index] + A = Image.open(A_path) params = get_params(self.opt, A.size) if self.opt.label_nc == 0: transform_A = get_transform(self.opt, params) -- cgit v1.2.3-70-g09d2