(self, crop_size, scale_size=None)
| 160 | |
| 161 | class GroupOverSample(object): |
| 162 | def __init__(self, crop_size, scale_size=None): |
| 163 | self.crop_size = crop_size if not isinstance(crop_size, int) else (crop_size, crop_size) |
| 164 | |
| 165 | if scale_size is not None: |
| 166 | self.scale_worker = GroupScale(scale_size) |
| 167 | else: |
| 168 | self.scale_worker = None |
| 169 | |
| 170 | def __call__(self, img_group): |
| 171 |