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/sequence_dataset.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) (limited to 'data/sequence_dataset.py') 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