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

Question

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

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

标签： pytorchdata-augmentationtorchvision

Answer 1

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