Skip to content

This project develops a CNN model to classify plastic waste into categories like "Bottles" or "Bags." Images are preprocessed, resized, and augmented for better training. A TensorFlow/Keras CNN with layers like Conv2D and MaxPooling is trained on the dataset and tested for accuracy, enabling automated waste sorting.

Notifications You must be signed in to change notification settings

Yasaswini-ch/Waste-Classification-Using-Convolution-Neural-Networking

Folders and files

NameName
Last commit message
Last commit date

Latest commit

ย 

History

16 Commits
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 

Repository files navigation

๐Ÿญ CNN Plastic Waste Classification

๐ŸŒ Overview

Plastic pollution is a major environmental challenge. This project leverages Convolutional Neural Networks (CNNs) to classify plastic waste, aiding in better waste management and recycling efforts.

๐Ÿ“‚ Dataset

๐Ÿ“ธ Total Images: 25,077
๐Ÿงช Training Data: 22,564 images (85%)
๐Ÿงช Test Data: 2,513 images (15%)
โ™ป๏ธ Categories: Organic and Recyclable

๐Ÿ—๏ธ Model Architecture

๐Ÿ”น Convolutional Layers - Extract key features from images
๐Ÿ”น Pooling Layers - Reduce dimensionality
๐Ÿ”น Fully Connected Layers - Classify images
๐Ÿ”น Activation Functions: ReLU & Softmax

๐ŸŽฏ Training

โš™๏ธ Optimizer: Adam
๐Ÿ“‰ Loss Function: Categorical Crossentropy
๐Ÿ“† Epochs: 25
๐Ÿ“ฆ Batch Size: 32

๐Ÿš€ How to Run

1๏ธโƒฃ Clone the repository:

git clone https://github.com/yasaswini-ch/CNN-Plastic-Waste-Classification
cd CNN-Plastic-Waste-Classification

2๏ธโƒฃ Install dependencies:

pip install -r requirements.txt

3๏ธโƒฃ Run the Streamlit app:

streamlit run app.py

๐Ÿ› ๏ธ Technologies Used

๐ŸŸก Python
๐ŸŸก TensorFlow/Keras
๐ŸŸก OpenCV
๐ŸŸก NumPy
๐ŸŸก Pandas
๐ŸŸก Matplotlib
๐ŸŸก Streamlit

๐Ÿ”ฎ Future Scope

โœจ Expand dataset with more waste categories
โœจ Improve accuracy with Transfer Learning
โœจ Deploy as a mobile application

๐Ÿ“œ License

๐Ÿ“ This project is licensed under the MIT License.


๐Ÿš€ Contribute to the project and help build a cleaner future! ๐ŸŒฑ

About

This project develops a CNN model to classify plastic waste into categories like "Bottles" or "Bags." Images are preprocessed, resized, and augmented for better training. A TensorFlow/Keras CNN with layers like Conv2D and MaxPooling is trained on the dataset and tested for accuracy, enabling automated waste sorting.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published