mingzailao / TFS-toolbox

Tensorflow Surgery Toolbox

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

TFS

Try to build some tools for Tensorflow

Setup

  • install python setup.py develop
  • uninstall python setup.py develop -u

Notebook

The python notebook for CNN visualization. The data is provided here:

NOTE: you need to download the data and change the name referenced in notebook file.

TODO

  • LRN layer should also have visualization pass. (in caffe, the author use the backward pass. Actually, in our experiment, whether adding it doesn't matter much)

About

Tensorflow Surgery Toolbox

License:MIT License


Languages

Language:Jupyter Notebook 87.0%Language:Python 12.0%Language:C++ 0.9%Language:Shell 0.1%