| 27 | |
| 28 | |
| 29 | class PairRandomResizedCrop(transforms.RandomResizedCrop): |
| 30 | def forward(self, imgs): |
| 31 | i, j, h, w = self.get_params(imgs[0], self.scale, self.ratio) |
| 32 | return [F.resized_crop(img, i, j, h, w, self.size, self.interpolation, antialias=self.antialias) for img in imgs] |
| 33 | |
| 34 | |
| 35 | class PairToTensor(transforms.ToTensor): |