mx.symbol.SpatialTransformer

Description

SpatialTransformer:Applies a spatial transformer to input feature map.

Usage

mx.symbol.SpatialTransformer(...)

Arguments

Argument

Description

data

NDArray-or-Symbol.

Input data to the SpatialTransformerOp.

loc

NDArray-or-Symbol localisation net, the output dim should be 6 when transform_type is affine. You shold initialize the weight and bias with identity tranform.

target.shape

Shape(tuple), optional, default=[0,0].

output shape(h, w) of spatial transformer: (y, x)

transform.type

{‘affine’}, required.

transformation type

sampler.type

{‘bilinear’}, required.

sampling type

cudnn.off

boolean or None, optional, default=None whether to turn cudnn off

name

string, optional.

Name of the resulting symbol.

Value

out The result mx.symbol