bhavanajain / spatial-transformer-networks

Tensorflow models for Spatial Transformer Networks

Spatial Transformer Network

The Spatial Transformer Network [1] allows the spatial manipulation of data within the network.

A Spatial Transformer Network implemented in Tensorflow 1.3.0 and based on [2].

Architecture

Experiments

We created RTS MNIST dataset by applying rotation, translation and scaling transformations on the original MNIST data. The dataset can be found here

All experiments were run in Tensorflow 1.3.0 on a GPU machine. Extensive results included the generated samples can be found here. We have also created videos for the samples generated by each model.

References

[1] Jaderberg, Max, et al. "Spatial Transformer Networks." arXiv preprint arXiv:1506.02025 (2015)

[2] https://github.com/daviddao/spatial-transformer-tensorflow