Skip to content

tomsoch/TransferLearning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 

Repository files navigation

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published