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) – 像素外推方法。默认：”replicate”。目前支持 “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。

上一页

megengine.functional.nn.remap

下一页

megengine.functional.nn.warp_perspective