transfer-learning-with-tensorflow

retrain_v5.py is a customised version of retrain.py for training on image data

Customizations Done

• Added 2 dense(Fully connected) layers followed by a SoftMax layer.
• Added dropout to both the dense layers.
• Added batch_normalization for dense layers
• Replaced Gradient-descent with Gradient-descent with Momentum
• Added live streamed confusion Matrix