Deep Learning Based Custom Image Multi-Class Classifier
This report details the design, and implementation of the multi-class image classifier using deep learning approach in Python with the help of TensorFlow. The data for this study were obtained from Kaggle and pre-processed and augmented through some transformations to make the final models more resilient. The model was trained using early stopping to avoid overfitting and the final model displayed a validation accuracy of 100%. In addition, the confusion matrix and training history plots show the remarkable accuracy of the model under consideration. This project demonstrates how the use of transfer learning and data augmentation enhances the performance of a model for the task of image classification.
Publication Details
- Journal
- Zenodo (CERN European Organization for Nuclear Research)
- Published
- 2024-12-17
- DOI
- https://doi.org/10.5281/zenodo.15678559
- Primary Topic
- Digital Imaging for Blood Diseases
- Type
- article
- Field-Weighted Citation Impact
- 0.00