Skip to content

Latest commit

 

History

12 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 

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

0 stars

Watchers

1 watching

Forks

Releases

Packages

Contributors

Languages