Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/zehaos/tfstn
Pure Tensorflow Implementation of Spatial Transformer Networks. TF API 1.0. Good visualization. Easy to be intergrated.
https://github.com/zehaos/tfstn
Last synced: 3 months ago
JSON representation
Pure Tensorflow Implementation of Spatial Transformer Networks. TF API 1.0. Good visualization. Easy to be intergrated.
- Host: GitHub
- URL: https://github.com/zehaos/tfstn
- Owner: Zehaos
- Created: 2017-04-07T09:12:05.000Z (almost 8 years ago)
- Default Branch: master
- Last Pushed: 2017-04-16T07:47:04.000Z (almost 8 years ago)
- Last Synced: 2023-10-20T10:09:09.550Z (about 1 year ago)
- Language: Python
- Homepage:
- Size: 334 KB
- Stars: 6
- Watchers: 4
- Forks: 4
- Open Issues: 1
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# 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](https://github.com/Zehaos/learn-tensorflow/blob/master/make_tfrecord.py)
### Train
```
python train.py
```
### Visualize
```
tensorboard --logdir=/tmp/zehao/logs/STN/train
```
## TODO
- [x] Train
- [x] Visualization
- [x] Transform input images
- [x] Different learning between stn&cnn
- [x] Tfrecord pipeline
- [ ] Intergrate into face recognition task
## Reference
[Spatial Transformer Networks](https://arxiv.org/pdf/1506.02025.pdf)
[IC-STN](https://github.com/ericlin79119/IC-STN)