如何在 PyTorch 中拉伸和挤压图像？

Question

我正在研究 MNIST 数字数据并寻找一种方法来拉伸或压缩图像。数据的形状为 (28,28)，我希望最终图像具有相同的形状，但可以拉伸或压缩。

有没有办法使用torchvision.transforms 或使用任何其他方法来做到这一点？

Answer 1

我想你正在寻找transforms.RandomAffine。
可以通过缩放来拉伸和挤压图像。