Waste-or-Garbage-Classification-Using-Deep-Learning

KO | EN

gitlite — search

#javascript #python #hacktoberfest #react #ai #typescript #llm #go #golang #android #machine-learning #rust #deep-learning #linux

Waste-or-Garbage-Classification-Using-Deep-Learning

(★ 62)

This model is created using pre-trained CNN architecture (VGG16 and RESNET50) via Transfer Learning that classifies the Waste or Garbage material (class labels =7) for recycling.

Waste-or-Garbage-Classification-Using-Deep-Learning Latest Version Download

Download Latest Version (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# Problem Statement
The problem is preety much stright forward, we all are famelier with Garbage and waste material which is very harmful for our society.if we talk about amount of waste then the world almost generates at least 5 million tons of waste per day and this number is still increasing day by day that's why we need to aware about waste. This model which help us to classify waste with 7 different waste materials and it will show you the details of that particular waste materials. This will help to raise awareness for people to reduce and recycle waste.

## Overview
- There are 'cardboard', 'compost', 'glass', 'metal', 'paper', 'plastic', 'trash' total 7 different types of waste materials which are use for recycling.
- Here i have 2187 images belonging 7 classes.
- Here i have trained dataset using VGG16,RESNET50 model via using Transfer Learning technique of CNN for classification.
- Here i have trained this model till 5 epochs and i got 41.31% accuracy on training data and 43% on testing data. Since my computer can not aford more epochs at this time due unavailbility of GPU. if you have good ammount of gpu you can train more than 50 epochs if you want to improve ac curacy.

---
## Introduction of Tranfer Learning and Fine Tuning

 <img src = './visualization_src/VGG16/2_f.jpg' align = 'center'>

---

 <img src = './visualization_src/VGG16/1_f.jpg' align = 'center'>

---
## VGG16 Architeture using Tranfer Learning
### Loss Plot of VGG16

 <img src = './visualization_src/VGG16/LossVal_loss.png' align = 'center'>

### Accuracies Plot of VGG16

 <img src = './visualization_src/VGG16/AccVal_acc.png' align = 'center'>

---
## ResNet50 Architeture using Tranfer Learning
### Loss Plot of ResNet50

 <img src = './visualization_src/ResNet50/ResNet_Loss.png' align = 'center'>

### Accuracies Plot of ResNet50

 <img src = './visualization_src/ResNet50/ResNet50_Accuracy.png' align = 'center'>

---

## Improve VGG-16 using Tranfer Learning
### Loss and Acuuracy Plot of VGG-16 with TF

 <img src = './visualization_src/VGG16/train_TF.png' align = 'center'>

## Improve VGG-16 using Fine Tuning
### Loss and Acuuracy Plot of VGG-16 with FT

 <img src = './visualization_src/VGG16/FT.png' align = 'center'>

---

## Evaluation Matrix of VGG16 and ResNet50 Model

| Model Name       |Test Accuracy | Epochs | For improve accuracy |
| ---------------- | ------------- | ------ | -------------------- |
| VGG-16 with TL    |     43.03%    |   5    |    Set 100 Epochs    |
| ResNet-50 with TL |     29.78%    |   5    |    Set 100 Epochs    |
| Improvement of VGG-16 with TL |     72.5%    |   20    |    Tune more HT    |
| Improvement of VGG-16 with TL+FT |     80.8%    |   25    |    Tune more HT    |

Waste-or-Garbage-Classification-Using-Deep-Learning

Waste-or-Garbage-Classification-Using-Deep-Learning Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

Waste-or-Garbage-Classification-Using-Deep-Learning

Waste-or-Garbage-Classification-Using-Deep-Learning Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

Download README