Image-Classification-using-Convolutional-Neural-Networks #811
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Pull Request for DL-Simplified 💡
Issue #730
Issue Title: Image Classification using Convolutional Neural Networks
Info about the related issue (Aim of the project): The goal of this project is to implement and compare the performance of various deep learning models, including LeNet-5, MobileNet, ResNet50, Simple CNN, and VGG16, for image classification tasks. By training these models on datasets like CIFAR-10 and MNIST, the project aims to evaluate and analyze their respective accuracies, loss values, and generalization capabilities. This comparative study will highlight the strengths and weaknesses of each architecture and provide insights into their effectiveness for practical image classification applications. The findings will guide the selection of the most suitable model architecture for robust image classification.
Name: Utsav Singhal
GitHub ID: UTSAVS26
Email ID: [email protected]
Identify yourself: SSoC 2024 participant
Closes: #730
Describe the add-ons or changes you've made 📃
This project involves the implementation and comparison of five deep learning models: LeNet-5, MobileNet, ResNet50, Simple CNN, and VGG16. Each model is designed for image classification tasks and has been tested on datasets like CIFAR-10 and MNIST. The models were trained with data augmentation to improve generalization and their performances were evaluated based on metrics such as accuracy, precision, recall, and F1-score. The project aims to identify which model best balances accuracy and generalization, offering insights into the advantages of different architectural features like residual connections and depthwise separable convolutions in deep learning.
Type of change ☑
What sort of change have you made:
How Has This Been Tested? ⚙
The project involved evaluating five deep learning models—LeNet-5, MobileNet, ResNet50, Simple CNN, and VGG16—for image classification tasks. The models were trained on the CIFAR-10 and MNIST datasets, with the data split into training, validation, and test sets. Training data underwent extensive augmentation to enhance model generalization. Each model was trained using categorical cross-entropy loss and the Adam optimizer, with callbacks for early stopping, best model checkpointing, and learning rate reduction. Performance was assessed on test sets using accuracy, precision, recall, F1-score, and confusion matrices. Results were visualized through accuracy curves and detailed classification reports. This comprehensive evaluation aimed to identify the most effective architecture for robust image classification.
Checklist: ☑