Warning: work in progress..

Transfer Learning for Tensorflow

• Object Detection based on TensorBox - GoogleNet/Overfeat/Rezoom
• Classification based on Tensorflow - InceptionV3

Setup

pip -r requirements.txt
cd transferflow/object_detection/utils; make; cd ../../../

Unit Tests

python test/object_detection_test.py

Todo

• Make threads shut down gracefully
• Refactor TensorBox originated codebase
• Improve model persistence
• Untangle rectangle stitching and results from the image rendering
• Bring in classification retrain code from TF
• Add unit tests for Classification
• Refactor
• Allow for more fine grained training controls + unit tests
• Experiment with different base models