Skip to content

yarinbnyamin/garbage_classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 

Repository files navigation

Garbage Classification

This work is on Kaggle Garbage Classification dataset which has 15,150 images from 12 different classes. The different classes are paper, cardboard, biological, metal, plastic, green glass, brown glass, white glass, clothes, shoes, batteries, and trash.

In this work, I trained EfficientNet-B0 with 86% Accuracy. See more at garbage_classification_efficientnet_b0.ipynb Open In Colab

About

Transfer learning to garbage classification using PyTorch and EfficientNet

Topics

Resources

License

Stars

Watchers

Forks