https://github.com/mpolinowski/tf-flowers

Get MobilNet from Tensorflow Hub and retrain it on custom dataset
https://github.com/mpolinowski/tf-flowers

image-classification mobilenetv2 tensorflow-hub tensorflow2

Last synced: about 2 months ago
JSON representation

Get MobilNet from Tensorflow Hub and retrain it on custom dataset

• Host: GitHub
• URL: https://github.com/mpolinowski/tf-flowers
• Owner: mpolinowski
• Created: 2022-12-20T05:02:02.000Z (over 3 years ago)
• Default Branch: master
• Last Pushed: 2022-12-20T05:02:06.000Z (over 3 years ago)
• Last Synced: 2025-03-23T13:14:13.791Z (about 1 year ago)
• Topics: image-classification, mobilenetv2, tensorflow-hub, tensorflow2
• Language: Python
• Homepage: https://mpolinowski.github.io/docs/IoT-and-Machine-Learning/ML/2022-12-19-tf-hub/2022-12-19
• Size: 679 KB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

          
# Tensorflow Hub

Trying out [TensorFlow Hub](https://www.tensorflow.org/hub) to download the feature extraction layer of MobileNet and retrain with the [flower dataset provided by Tensorflow](https://storage.googleapis.com/download.tensorflow.org/example_images/flower_photos.tgz):

* `testing_mobilenet.py`: Test the untrained model

* `training_mobilenet.py`: Train the model and testing

![Tensorflow Hub - MobileNet](./Tensorflow_Transfer_Learning_05.png)