mpolinowski/tf-flowers

Get MobilNet from Tensorflow Hub and retrain it on custom dataset

Tensorflow Hub

Trying out TensorFlow Hub to download the feature extraction layer of MobileNet and retrain with the flower dataset provided by Tensorflow:

• testing_mobilenet.py: Test the untrained model
• training_mobilenet.py: Train the model and testing