megengine.functional.vision.warp_affine¶

warp_affine(inp, weight, out_shape, border_mode='REPLICATE', border_val=0, format='NHWC', imode='LINEAR')[源代码]¶

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

参数

inp (Tensor) – 输入图像。
weight (Tensor) – 权重 Tensor。
out_shape – 输出张量的形状。
border_mode – 像素外推方法。默认值： “ WRAP”。当前支持“ CONSTANT”，“ REFLECT”，“ REFLECT_101”，“ ISOLATED”，“ WRAP”，“ REPLICATE”，“ TRANSPARENT”。
border_val – 边界填充值。默认：0
format – 由于历史因素，默认格式是“NHWC”，也支持“NCHW”格式。
imode – 插值方法。可以是“ LINEAR”，“ NEAREST”，“ CUBIC”，“ AREA”。默认值：“ LINEAR”。

返回

输出张量。

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