(self, crop_size, scale_size=None)
| 97 | |
| 98 | class GroupOverSample(object): |
| 99 | def __init__(self, crop_size, scale_size=None): |
| 100 | self.crop_size = crop_size if not isinstance(crop_size, int) else (crop_size, crop_size) |
| 101 | |
| 102 | if scale_size is not None: |
| 103 | self.scale_worker = GroupScale(scale_size) |
| 104 | else: |
| 105 | self.scale_worker = None |
| 106 | |
| 107 | def __call__(self, img_group): |
| 108 |