TFSTN

This is a pure tensorflow implementation of Spatial Transformer Networks.

Features:

• TF API 1.0.0
• Good visualization.
• Easy to be intergrated.
• Tfrecord input pipeline

Graph

Result

Usage

Make mnist tfrecord

script

Train

python train.py

Visualize

tensorboard --logdir=/tmp/zehao/logs/STN/train

TODO

• Train
• Visualization
• Transform input images
• Different learning between stn&cnn
• Tfrecord pipeline
• Intergrate into face recognition task

Reference

Spatial Transformer Networks

IC-STN