megengine.functional.vision.warp_affine

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

对按批组织的二维图片进行仿射变换。

参数
  • inp – 输入图像。

  • mat(batch, 2, 3) transformation matrix.

  • out_shape – 输出张量的形状。

  • border_mode – pixel extrapolation method. Default: “wrap”. Currently “constant”, “reflect”, “reflect_101”, “isolated”, “wrap”, “replicate”, “transparent” are supported.

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

  • format – “NHWC” as default based on historical concerns, “NCHW” is also supported. Default: “NHWC”.

  • interp_mode – interpolation methods. Could be “linear”, “nearest”, “cubic”, “area”. Default: “linear”.

返回

输出张量。

注解

这里列出了参数的所有可用选项,但这并不意味着您可以使用所有的组合。在不同的平台上,支持不同的组合。