pk00095 / transfer-learning-with-tensorflow Public

Notifications You must be signed in to change notification settings
Fork 4
Star 1

A customised version of retrain.py for training on image data

1 star 4 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 20 Commits
LICENSE		LICENSE
README.md		README.md
customization_details.ipynb		customization_details.ipynb
deep_network.py		deep_network.py
retrain_v5.py		retrain_v5.py

Repository files navigation

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

About

A customised version of retrain.py for training on image data

python deep-learning tensorflow image-classification tensorflow-experiments

Report repository

Releases

No releases published

Packages

No packages published

Languages