affine_grid
paddle.fluid.layers.
affine_grid
(theta, out_shape, name=None)[源代码]
该OP用于生成仿射变换前后的feature maps的坐标映射关系。在视觉应用中,根据该OP得到的映射关系,将输入feature map的像素点变换到对应的坐标,就得到了经过仿射变换的feature map。
- 参数:
- theta (Variable) - Shape为
[batch_size, 2, 3]
的Tensor,表示batch_size个2X3
的变换矩阵。数据类型支持float32,float64。- out_shape (Variable | list | tuple) - 类型可以是1-D Tensor、list或tuple。用于表示在仿射变换中的输出的shape,其格式
[N, C, H, W]
,分别为输出feature map的batch size、channel数量、高和宽。数据类型支持int32。
- name (None|str) – 该参数供开发人员打印调试信息时使用,具体用法请参见 :ref:api_guide_Name ,默认值为None。
返回: Variable。Shape为 [N, H, W, 2]
的4-D Tensor,表示仿射变换前后的坐标的映射关系。其中,N、H、W分别为仿射变换中输出feature map的batch size、高和宽。 数据类型与 theta
一致。
返回类型:Variable
代码示例:
- import paddle.fluid as fluid
- theta = fluid.layers.data(name="x", shape=[2, 3], dtype="float32")
- out_shape = fluid.layers.data(name="y", shape=[-1], dtype="float32")
- data = fluid.layers.affine_grid(theta, out_shape)
- # or
- data = fluid.layers.affine_grid(theta, [5, 3, 28, 28])