我想在PyTorch中使用变换矩阵执行图像变换。我之前的代码是在TensorFlow中实现的，所以我想知道是否有适用于tfa.image.transform的PyTorch等效函数？谢谢你。
0mkxixxg1#
投影图像转换可以使用kornia库的kornia.geometry.transform.warp_affine函数完成。API可能不是tfa.image.transform的替代品。有关详细信息，请参阅https://kornia.readthedocs.io/en/latest/geometry.transform.html#kornia-geometry-transform下面是kornia变换的图像。
kornia.geometry.transform.warp_affine
m2xkgtsf2#
我认为您正在寻找torch.nn.functional.grid_sample来根据新的空间坐标对图像（或2D特征Map）进行采样，当与torch.nn.functional.affine_grid一起使用时，它将对输入图像/特征Map执行仿射空间变换。
torch.nn.functional.grid_sample
torch.nn.functional.affine_grid
2条答案
按热度按时间0mkxixxg1#
投影图像转换可以使用kornia库的
kornia.geometry.transform.warp_affine函数完成。API可能不是tfa.image.transform的替代品。有关详细信息，请参阅https://kornia.readthedocs.io/en/latest/geometry.transform.html#kornia-geometry-transform下面是kornia变换的图像。
m2xkgtsf2#
我认为您正在寻找
torch.nn.functional.grid_sample来根据新的空间坐标对图像（或2D特征Map）进行采样，当与torch.nn.functional.affine_grid一起使用时，它将对输入图像/特征Map执行仿射空间变换。