(self, imgs: list, targets: list)
| 316 | |
| 317 | class MotRandomCrop(RandomCrop): |
| 318 | def __call__(self, imgs: list, targets: list): |
| 319 | ret_imgs = [] |
| 320 | ret_targets = [] |
| 321 | region = T.RandomCrop.get_params(imgs[0], self.size) |
| 322 | for img_i, targets_i in zip(imgs, targets): |
| 323 | img_i, targets_i = crop(img_i, targets_i, region) |
| 324 | ret_imgs.append(img_i) |
| 325 | ret_targets.append(targets_i) |
| 326 | return ret_imgs, ret_targets |
| 327 | |
| 328 | class FixedMotRandomCrop(object): |
| 329 | def __init__(self, min_size: int, max_size: int): |