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)` 的转换矩阵border_mode (
str
) – 边界处理方式。默认为”wrap”。 现在支持”constant”, “reflect”, “reflect_101”, “isolated”, “wrap”, “replicate”, “transparent”。border_val (
float
) – 边界填充值。 默认:0format (
str
) – 因为历史原因,默认使用 “NHWC”,也支持 “NCHW”。默认值:“NHWC”interp_mode (
str
) – 插值方式。可以是 ”linear“, “nearest”, ”cubic“,”area“。默认是:“linear”。
- 返回类型
- 返回
输出张量。
注解
这里列举了所有的可用的参数选项,但是并不是说支持这些参数选项的任意组合。在不同的平台上,支持不同的参数组合。
warp_affine
只支持正向推理, 如果需要反向推理请参考warp_perspective
。