数据增广时 input 和 target 保持相同变换
图像分割任务中,input 和 target 都是图片,此时做数据增广时如果用到随机变换,例如随机翻转和旋转,则需要保证两者随机值相同,否则分割 label 就不准确。
例如变换:
self.transform = transforms.Compose([
transforms.RandomHorizontalFlip(),
transforms.RandomRotation(90),
transforms.CenterCrop(480),
transforms.ToTensor(),
transforms.ToPILImage()
])
在使用时如果直接
image = self.transform(image)
bulb = self.transform(bulb)
则输入图片和 label 会产生不同的翻转/旋转角,造成数据不对应的问题。