For a TF-2.0 rewrite visit: https://github.com/oarriaga/paz Spatial transformer networks Implementation of spatial transformer networks in keras 2 using tensorflow backend.
