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