如何使用 PyTorch 变换对输入/输出图像对应用相同的变换？

Question

我想通过 PyTorch 构建一个深度学习模型来进行一些图像增强。模型的输入和输出是相同大小的图像。

我想使用 PyTorch 的

torchvision.transforms

库进行数据增强。许多有用的变换的性质是随机的，所以我不知道如何在输入/输出对上应用相同的变换。我的意思是，应用于输入图像的相同变换必须应用于相应的输出图像。

Answer 1

您最好使用albumentation进行图像增强，您可以将图像和掩模一起提供给增强功能。

否则，您可以随机生成输入参数（例如旋转角度），然后调用与其对应的确定性函数，例如本例中的 F.rotate。