This project involves building an image classifier using a custom model that resembles ResNet 34 and also employing transfer learning using a pretrained MobileNet network. Learning rate range test is employed to determine the apt learning rate. Data augmentaion is used to regularize the model. The models are trained with a train set of size 4000 images and validated on a validation set of size 1000 images. I have also tried building a classifier using SVM trained with SIFT features in bag of visual words representation.
forked from VishalRishi23/Image-classifier
-
Notifications
You must be signed in to change notification settings - Fork 0
viswa2811/Image-classifier
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published
Languages
- Jupyter Notebook 100.0%