This repository contains code for a deep learning model to classify various types of garbage using a dataset available on Kaggle. The goal is to develop a robust model capable of accurately identifying and classifying garbage types to support waste management and recycling efforts.
The dataset used in this project is available at Kaggle Garbage Classification Dataset. It includes images of different types of waste and is organized into several categories.
data/: Contains the dataset and any preprocessed data.
notebooks/: Jupyter notebooks for exploratory data analysis and model training.
scripts/: Python scripts for data preprocessing, model training, and evaluation.
models/: Trained model files and checkpoints.
requirements.txt: List of Python packages required for this project.
README.md: This file.
Getting Started
Prerequisites Ensure you have the following Python packages installed:
TensorFlow or PyTorch
NumPy
Pandas
OpenCV
Matplotlib
Scikit-learn
Special thanks to Ahmet Haydar Ornek for his inspiring work on Transfer Learning models. This project is inspired by his approach and methodology, though it has been adapted and customized to suit this specific task. Kaggle for providing the dataset. TensorFlow and PyTorch for the deep learning frameworks used.