我想使用 PyTorch 中的变换矩阵执行图像变换。我之前的代码是在TensorFlow中实现的,所以我想知道是否有tfa.image.transform的PyTorch等效函数?
谢谢。
最佳答案
可以使用 kornia 库的 kornia.geometry.transform.warp_affine
函数来完成投影图像转换。该 API 可能不会完全替代 tfa.image.transform。详情请参阅https://kornia.readthedocs.io/en/latest/geometry.transform.html#kornia-geometry-transform
下面是 kornia 变换的图像。
关于python - tfa.image.transform 的 PyTorch 等效函数,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/74938933/