|
| 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 | + |
| 44 | + |
| 45 | +### Model 2 (Basic CNN Model) Performance Graphs |
| 46 | + |
| 47 | + |
| 48 | +### Model 3 (VGG16 Model) Performance Graphs |
| 49 | + |
| 50 | + |
| 51 | +### Model 4 (ResNet Model) Performance Graphs |
| 52 | + |
| 53 | + |
| 54 | +### Model 5 (MobileNetV2 Model) Performance Graphs |
| 55 | + |
| 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