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.
-
Notifications
You must be signed in to change notification settings - Fork 0
MaheshDY/ML_Transfer_learning
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
Image classification based on Transfer Learning and pre-trained weights.
Topics
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published