Torch Implementation of Volumetric Spatial Transformer Network which we used in ALIGNet. Code borrows heavily from the PTN torch implementation and 2D STN torch implementation.
git clone https://github.com/ranahanocka/stn3dtorch.git
cd stn3dtorch
luarocks make stn3dtorch-scm-1.rockspecif everything went well, this import should work:
require 'volstn'