Pytorch 变换的两个图像上的相同随机裁剪

Question

我正在尝试将两张图像输入网络，并且我想在这两张图像之间进行相同的转换。 transforms.Compose() 一次拍摄一张图像并生成彼此独立的输出，但我想要相同的转换。我为 hflip() 编写了自己的代码，现在我有兴趣获得随机裁剪。有没有办法不编写自定义函数就可以做到这一点？

Answer 1

我会使用这样的解决方法 - 使我自己的作物 class 继承自 RandomCrop，重新定义 call with

    …
        if self.call_is_even :
            self.ijhw = self.get_params(img, self.size)
        i, j, h, w = self.ijhw
        self.call_is_even = not self.call_is_even

而不是

    i, j, h, w = self.get_params(img, self.size)

这个想法是在奇数调用时抑制随机化器