(self, imgs, targets)
| 466 | |
| 467 | class MotRandomHorizontalFlip(RandomHorizontalFlip): |
| 468 | def __call__(self, imgs, targets): |
| 469 | if random.random() < self.p: |
| 470 | ret_imgs = [] |
| 471 | ret_targets = [] |
| 472 | for img_i, targets_i in zip(imgs, targets): |
| 473 | img_i, targets_i = hflip(img_i, targets_i) |
| 474 | ret_imgs.append(img_i) |
| 475 | ret_targets.append(targets_i) |
| 476 | return ret_imgs, ret_targets |
| 477 | return imgs, targets |
| 478 | |
| 479 | |
| 480 | class RandomResize(object): |