diff options
Diffstat (limited to 'data')
| -rw-r--r-- | data/recursive_dataset.py | 2 | ||||
| -rw-r--r-- | data/sequence_dataset.py | 12 |
2 files changed, 7 insertions, 7 deletions
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) |