Spatial Transformer Network

Get this model


This model enables training a Spatial Transformer Network on a cluttered version of the MNIST dataset.

A Spatial Transformer Network allows the spatial manipulation of data within the network.


It can be inserted into existing convolutional architectures, giving neural networks the ability to actively spatially transform feature maps, conditional on the feature map itself, without any extra training supervision or modification to the optimisation process [1].

Model Metadata

Domain Application Industry Framework Training Data Input Data Format
Vision Image Classification General TensorFlow Cluttered MNIST Image (RGB/HWC)



Component License Link
Model GitHub Repository Apache 2.0 LICENSE
Model Code (3rd party) Apache 2.0 TensorFlow Models
Data MIT Cluttered MNIST

Options available for training this model