megengine.functional.nn.warp_affine

warp_affine(inp, mat, out_shape, border_mode='replicate', border_val=0.0, format='NHWC', interp_mode='linear')[源代码]

二维图像上的批量仿射变换,这里的仿射变换是指二维坐标间的线性变换。

参数
  • inp (Tensor) – 输入图像。

  • mat (Tensor) – 变换矩阵,形状为`(batch, 2, 3)`

  • out_shape (Union[Tuple[int, int], int, Tensor]) – 输出张量的形状。

  • border_mode (str) – 边界处理方式。默认为”wrap”。 现在支持”constant”, “reflect”, “reflect_101”, “isolated”, “wrap”, “replicate”, “transparent”。

  • border_val (float) – 边界填充值。 默认:0

  • format (str) – 因为历史原因,默认使用 “NHWC”,也支持 “NCHW”。默认值:“NHWC”

  • interp_mode (str) – 插值方式。可以是 ”linear“, “nearest”, ”cubic“,”area“。默认是:“linear”。

返回类型

Tensor

返回

输出张量。

注解

这里列举了所有的可用的参数选项,但是并不是说支持这些参数选项的任意组合。在不同的平台上,支持不同的参数组合。warp_affine 只支持正向推理, 如果需要反向推理请参考 warp_perspective