Built a model for image classification using CNN from scratch i.e. without using any available library. 12 Layer model was created followed by cross entropy loss for SoftMax regression. Model was trained on Intel dataset, initially the accuracy was 56% for train and 48% for test. This was further improved by various optimization techniques, starting with tickling hyperparameters but didn’t get any considerable improvement in accuracy, followed by L2 regularization along with HE initialization which successfully boost the accuracy of model with 91% on training data and 89% on test data.
-
Notifications
You must be signed in to change notification settings - Fork 0
Prachivats/Object-Detection-using-CNN
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Image Classification using CNN
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published