| 36 | std=[0.229, 0.224, 0.225]) |
| 37 | |
| 38 | def parse_input_list(self, odgt, max_sample=-1, start_idx=-1, end_idx=-1): |
| 39 | if isinstance(odgt, list): |
| 40 | self.list_sample = odgt |
| 41 | elif isinstance(odgt, str): |
| 42 | self.list_sample = [json.loads(x.rstrip()) for x in open(odgt, 'r')] |
| 43 | |
| 44 | if max_sample > 0: |
| 45 | self.list_sample = self.list_sample[0:max_sample] |
| 46 | if start_idx >= 0 and end_idx >= 0: # divide file list |
| 47 | self.list_sample = self.list_sample[start_idx:end_idx] |
| 48 | |
| 49 | self.num_sample = len(self.list_sample) |
| 50 | assert self.num_sample > 0 |
| 51 | print('# samples: {}'.format(self.num_sample)) |
| 52 | |
| 53 | def img_transform(self, img): |
| 54 | # 0-255 to 0-1 |