(self, dataroots, load_size=64)
| 9 | |
| 10 | class TwoAFCDataset(BaseDataset): |
| 11 | def initialize(self, dataroots, load_size=64): |
| 12 | if(not isinstance(dataroots,list)): |
| 13 | dataroots = [dataroots,] |
| 14 | self.roots = dataroots |
| 15 | self.load_size = load_size |
| 16 | |
| 17 | # image directory |
| 18 | self.dir_ref = [os.path.join(root, 'ref') for root in self.roots] |
| 19 | self.ref_paths = make_dataset(self.dir_ref) |
| 20 | self.ref_paths = sorted(self.ref_paths) |
| 21 | |
| 22 | self.dir_p0 = [os.path.join(root, 'p0') for root in self.roots] |
| 23 | self.p0_paths = make_dataset(self.dir_p0) |
| 24 | self.p0_paths = sorted(self.p0_paths) |
| 25 | |
| 26 | self.dir_p1 = [os.path.join(root, 'p1') for root in self.roots] |
| 27 | self.p1_paths = make_dataset(self.dir_p1) |
| 28 | self.p1_paths = sorted(self.p1_paths) |
| 29 | |
| 30 | transform_list = [] |
| 31 | transform_list.append(transforms.Scale(load_size)) |
| 32 | transform_list += [transforms.ToTensor(), |
| 33 | transforms.Normalize((0.5, 0.5, 0.5),(0.5, 0.5, 0.5))] |
| 34 | |
| 35 | self.transform = transforms.Compose(transform_list) |
| 36 | |
| 37 | # judgement directory |
| 38 | self.dir_J = [os.path.join(root, 'judge') for root in self.roots] |
| 39 | self.judge_paths = make_dataset(self.dir_J,mode='np') |
| 40 | self.judge_paths = sorted(self.judge_paths) |
| 41 | |
| 42 | def __getitem__(self, index): |
| 43 | p0_path = self.p0_paths[index] |
nothing calls this directly
no test coverage detected