tomsoch / TransferLearning

Transfer learning script that operates on pre-trained SSD MobileNet models

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

TransferLearning

Transfer Learning script that operates on an SSD MobileNet V1 and V2 model pre-trained on a COCO image dataset

It lets users perform a transfer learning process on its own dataset of images

The training process can be watched with the Tensorboard visualization tool

tensorboard_cube

Images have been labelled using graphical image annotation tool - LabelImg

image

Example result of a trained model that used a private dataset of Rubik's cube:

pobrane (1)

The newly trained model can be used with the ObjectToSpeech application to perform real-time detection https://github.com/tomsoch/ObjectToSpeech

image

About

Transfer learning script that operates on pre-trained SSD MobileNet models


Languages

Language:Jupyter Notebook 99.9%Language:Python 0.1%