transform a group of images independently
| 14 | |
| 15 | |
| 16 | class StackTransform(object): |
| 17 | """transform a group of images independently""" |
| 18 | def __init__(self, transform): |
| 19 | self.transform = transform |
| 20 | |
| 21 | def __call__(self, imgs): |
| 22 | return torch.stack([self.transform(crop) for crop in imgs]) |
| 23 | |
| 24 | |
| 25 | class JigsawCrop(object): |