当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python mxnet.symbol.op.BilinearSampler用法及代码示例


用法:

mxnet.symbol.op.BilinearSampler(data=None, grid=None, cudnn_off=_Null, name=None, attr=None, out=None, **kwargs)

参数：

  • data：(Symbol) - 将数据输入到 BilinearsamplerOp。
  • grid：(Symbol) - BilinearsamplerOp.grid 的输入网格有两个通道：x_src、y_src
  • cudnn_off：(boolean or None, optional, default=None) - 是否关闭 cudnn
  • name：(string, optional.) - 结果符号的名称。

返回：

结果符号。

返回类型：

Symbol

将双线性采样应用于输入特征图。

双线性采样是[NIPS2015]“Spatial Transformer Networks”的关键。该算子的用法与OpenCV中的remap函数非常相似，只不过该算子具有向后传递函数。

给定 和 ，则输出由下式计算

, 枚举 中的所有空间位置， 表示双线性插值内核。边界点将用零填充。输出的形状将是(data.shape[0]、data.shape[1]、grid.shape[2]、grid.shape[3])。

操作符假设 具有“NCHW”布局，并且 已标准化为 [-1, 1]。

BilinearSampler经常与GridGenerator配合，为BilinearSampler生成采样网格。 GridGenerator 支持两种转换： affine 和 warp 。如果用户想设计一个CustomOp来操作 ，请首先参考GridGenerator的代码。

示例 1：

## Zoom out data two times
data = array([[[[1, 4, 3, 6],
                [1, 8, 8, 9],
                [0, 4, 1, 5],
                [1, 0, 1, 3]]]])

affine_matrix = array([[2, 0, 0],
                       [0, 2, 0]])

affine_matrix = reshape(affine_matrix, shape=(1, 6))

grid = GridGenerator(data=affine_matrix, transform_type='affine', target_shape=(4, 4))

out = BilinearSampler(data, grid)

out
[[[[ 0,   0,     0,   0],
   [ 0,   3.5,   6.5, 0],
   [ 0,   1.25,  2.5, 0],
   [ 0,   0,     0,   0]]]

示例 2：

## shift data horizontally by -1 pixel

data = array([[[[1, 4, 3, 6],
                [1, 8, 8, 9],
                [0, 4, 1, 5],
                [1, 0, 1, 3]]]])

warp_maxtrix = array([[[[1, 1, 1, 1],
                        [1, 1, 1, 1],
                        [1, 1, 1, 1],
                        [1, 1, 1, 1]],
                       [[0, 0, 0, 0],
                        [0, 0, 0, 0],
                        [0, 0, 0, 0],
                        [0, 0, 0, 0]]]])

grid = GridGenerator(data=warp_matrix, transform_type='warp')
out = BilinearSampler(data, grid)

out
[[[[ 4,  3,  6,  0],
   [ 8,  8,  9,  0],
   [ 4,  1,  5,  0],
   [ 0,  1,  3,  0]]]

相关用法


注：本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.symbol.op.BilinearSampler。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。