PyTorch version of spatial transformer network - GitHub
github.com › fxia22 › stnJun 17, 2017 · STN is the spatial transformer module, it takes a B*H*W*D tensor and a B*H*W*2 grid normalized to [-1,1] as an input and do bilinear sampling. AffineGridGen takes a B*2*3 matrix and generate an affine transformation grid. CylinderGridGen takes a B*1 theta vector and generate a transformation grid to remap equirectangular images along x axis.