(self, transform_type)
| 15 | self.jitter_param = jitter_param |
| 16 | |
| 17 | def parse_transform(self, transform_type): |
| 18 | if transform_type == 'ImageJitter': |
| 19 | method = add_transforms.ImageJitter(self.jitter_param) |
| 20 | return method |
| 21 | method = getattr(transforms, transform_type) |
| 22 | if transform_type == 'RandomSizedCrop': |
| 23 | return method(self.image_size) |
| 24 | elif transform_type == 'CenterCrop': |
| 25 | return method(self.image_size) |
| 26 | elif transform_type == 'Resize': |
| 27 | return method([int(self.image_size * 1.15), int(self.image_size * 1.15)]) |
| 28 | elif transform_type == 'Normalize': |
| 29 | return method(**self.normalize_param) |
| 30 | else: |
| 31 | return method() |
| 32 | |
| 33 | def get_composed_transform(self, aug=False): |
| 34 | if aug: |
no outgoing calls
no test coverage detected