Skip to content

Commit 1c321d4

Browse files
committed
Adding Waste Classification using DL project
1 parent 55d361e commit 1c321d4

9 files changed

+6673
-0
lines changed
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
# Waste Classification Using Deep Learning
2+
3+
The dataset used in this project is taken from the Kaggle website.
4+
5+
**Dataset Link:** [Waste Classification Data](https://www.kaggle.com/datasets/techsash/waste-classification-data)
6+
7+
This dataset consists of images of organic (O) and recyclable (R) waste. The images are labeled to provide a comprehensive dataset for training and evaluating machine learning models.
8+
9+
The task is to classify each waste image into one of the two predefined categories indicating either organic or recyclable waste.
Loading
Loading
Loading
Loading
Loading
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,92 @@
1+
# Waste Classification Using Deep Learning
2+
3+
**GOAL**
4+
5+
To classify images from the waste classification dataset using a deep learning approach.
6+
7+
**DATASET**
8+
9+
[Waste Classification Data](https://www.kaggle.com/datasets/techsash/waste-classification-data)
10+
11+
**DESCRIPTION**
12+
13+
The dataset contains images of organic (O) and recyclable (R) waste. The task is to classify these images using deep learning architectures.
14+
15+
**WHAT I DID**
16+
17+
First, I imported all the required libraries and the dataset for this project. I split the dataset into training, validation, and testing sets. Then I proceeded to build and evaluate the models.
18+
19+
I developed several deep learning models to classify the images. Initially, I used an Artificial Neural Network (ANN), followed by a Convolutional Neural Network (CNN). Both models did not yield satisfactory accuracy. I then used VGG16, ResNet, and MobileNetV2 architectures, with the latter providing better performance. Finally, I evaluated the performance of all the models in order to pick the optimal model.
20+
21+
**MODELS USED**
22+
23+
The models are:
24+
25+
1. Artificial Neural Network (ANN)
26+
2. Basic Convolutional Neural Network (CNN)
27+
3. VGG16 Model
28+
4. ResNet Model
29+
5. MobileNetV2 Model
30+
31+
**LIBRARIES NEEDED**
32+
33+
- tensorflow
34+
- matplotlib
35+
- opencv-python
36+
- numpy
37+
- random
38+
- shutil
39+
40+
**VISUALIZATION**
41+
42+
### Model 1 (ANN Model) Performance Graphs
43+
![Model 1 (ANN Model) performance graphs](../Images/ANN_Performance.png)
44+
45+
### Model 2 (Basic CNN Model) Performance Graphs
46+
![Model 2 (Basic CNN Model) performance graphs](../Images/CNN_Performance.png)
47+
48+
### Model 3 (VGG16 Model) Performance Graphs
49+
![Model 3 (VGG16 Model) performance graphs](../Images/VGG16_Performance.png)
50+
51+
### Model 4 (ResNet Model) Performance Graphs
52+
![Model 4 (ResNet Model) performance graphs](../Images/ResNet50_Performance.png)
53+
54+
### Model 5 (MobileNetV2 Model) Performance Graphs
55+
![Model 5 (MobileNetV2 Model) performance graphs](../Images/MobileNetV2_Performance.png)
56+
57+
**ACCURACIES**
58+
59+
| Model | Architecture | Accuracy in % (on testing data) |
60+
|--------------------|:---------------------------:|:------------------------------:|
61+
| Model 1 | ANN Model | 81.73 |
62+
| Model 2 | Basic CNN Model | 87.31 |
63+
| Model 3 | VGG16 Model | 90.65 |
64+
| Model 4 | ResNet Model | 76.88 |
65+
| Model 5 | MobileNetV2 Model | 90.85 |
66+
67+
**CONCLUSION**
68+
69+
After training and evaluating various deep learning models on the waste classification dataset, the results are as follows:
70+
71+
**Key Observations:**
72+
73+
1. **Artificial Neural Network (ANN) Model**: Achieved an accuracy of 81.73%. While it provided a decent baseline, it was outperformed by the other convolutional neural network-based models.
74+
75+
2. **Basic CNN Model**: Improved accuracy significantly to 87.31%, demonstrating the power of convolutional layers in image classification tasks.
76+
77+
3. **VGG16 Model**: Achieved an impressive accuracy of 90.65%. This pre-trained model showed strong performance, indicating the benefits of transfer learning.
78+
79+
4. **ResNet Model**: Obtained an accuracy of 76.88%, which was lower than expected. This might be due to overfitting or insufficient fine-tuning.
80+
81+
5. **MobileNetV2 Model**: Achieved the highest accuracy of 90.85%. This lightweight model is particularly suitable for deployment in resource-constrained environments.
82+
83+
**Conclusion:**
84+
85+
The MobileNetV2 model emerged as the best-performing model for the waste classification task, with an accuracy of 90.85%. Its balance of high accuracy and computational efficiency makes it an excellent choice for practical applications. The VGG16 model also performed very well and could be a suitable alternative depending on the specific use case requirements.
86+
87+
Further improvements can be made by fine-tuning the models, exploring additional data augmentation techniques, and experimenting with different hyperparameters.
88+
89+
**Connect with Me**
90+
91+
- [LinkedIn](https://www.linkedin.com/in/barrenkala-veera-venkata-karthik-b58b9a285/)
92+
- [GitHub](https://github.com/Karthik110505)

0 commit comments

Comments
 (0)