abhisheks008 / DL-Simplified

Deep Learning Simplified is an Open-source repository, containing beginner to advance level deep learning projects for the contributors, who are willing to start their journey in Deep Learning. Devfolio URL, https://devfolio.co/projects/deep-learning-simplified-f013
https://quine.sh/repo/abhisheks008-DL-Simplified-499023976
MIT License
334 stars 296 forks source link

[Model Enhancement]: Colour Detection using DL #552

Closed Arihant-Bhandari closed 2 months ago

Arihant-Bhandari commented 2 months ago

Pull Request for DL-Simplified 💡

Issue Title : [Model Enhancement]: Colour Detection using DL

Closes: #529 [Model Enhancement]: Colour Detection using DL

Describe the add-ons or changes you've made 📃

i have enchanced the previous model by adding the new image loading function from keras.utils, Attention mechanism, using ELU instead of ReLU due to its better functionality, made the problem into a multi-class classification model with softmax implementation, used the optimizer Nadam with lr = 0.01, displayed the first 32 kernels based on the images, saved and included the best model as part of the PR.

Type of change ☑️

What sort of change have you made:

How Has This Been Tested? ⚙️

Previous Accuracy : 0.9

New metrics:

Accuracy: 0.94

Precision: 0.9565476190476192

Recall: 0.9514285714285714

F1 Score: 0.95005439005439

Confusion Matrix: [[6 0 0 0 1 0 0 0 0 0] [0 4 0 0 0 0 0 0 0 0] [0 0 7 0 0 0 0 0 0 0] [0 0 0 2 0 0 0 0 0 0] [0 0 0 0 6 0 0 0 0 0] [0 0 0 0 0 4 0 0 0 1] [0 0 1 0 0 0 6 0 0 0] [0 0 0 0 0 0 0 5 0 0] [0 0 0 0 0 0 0 0 2 0] [0 0 0 0 0 0 0 0 0 5]]

Checklist: ☑️

github-actions[bot] commented 2 months ago

Our team will soon review your PR. Thanks @Arihant-Bhandari :)

Arihant-Bhandari commented 2 months ago

@abhisheks008 hi, as you suggest in the last PR, i have made appropriate changes, if there is still something that needs changes, please let me know, i would make them.

thank you for your time reviewing this PR. Hope to hear from you soon.