Garbage-Classification

Autori

Questo progetto è stato realizzato da Giada Margarone e Miriana Russo

Overview

Lo scopo di questo progetto è quello di allenare un modello affinchè sappia classificare i rifiuti in 6 diverse classi: glass, paper, cardboard, plastic, metal, trash.

Dataset

È stato utilizzato un dataset kaggle Garbage Classification (≈2.5 k immagini, ritagliate 384x384). Il dataset viene automaticamente scaricato all'avvio del file main.ipynb.

Layout Progetto

project-garbage-classifier/
│
├── README.md          
├── requirements.txt   
│
├── database/
│   └── database.py    
│
├── src/
│   ├── pre_processing.py 
│   ├── classifier.py      
│   └── main.py           
│
└── .gitignore

Dependencies

Assicurati di avere git lfs installato.

Demo quick start

# 1. Clona la repository
git clone https://github.com/miryrusso/Garbage-Classification.git

# 2. Entra nella cartella
cd Garbage-Classification && git lfs pull

# 3. crea e attiva il virtual-env
python3 -m venv .venv && source .venv/bin/activate

# 4. installa le dipendenze
pip install -r requirements.txt

# 5. demo
cd demo
python3 demo.py

Demo.mp4

Relazione

La relazione finale del progetto è disponibile al seguente link