Skip to content

Image classification based on Transfer Learning and pre-trained weights.

Notifications You must be signed in to change notification settings

MaheshDY/ML_Transfer_learning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 

Repository files navigation

ML_Transfer_learning

The base_model.ipynb is the parent model which is trained on cats and dogs image classification dataset. The weights of this parent model are saved for later use. This parent model is used as a base model for the image classification of another horses or humans dataset by using the same weights and slight modification to the parent model like using regularization and data augumentation.

About

Image classification based on Transfer Learning and pre-trained weights.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published