garbage_classification

CV classification task for automatic garbage classification to make recycling tasks more efficient.

Dataset

Build a dataset of 7 classes of usual garbage: paper, cardboard, can, glass bottle, plastic bottle, plastic bag and battery using Flickr API. Each class contains 500 pictures.

Data processing

1. Resize
2. Normalization
3. Augmentation

Models

1. Baseline chosing: MLP; LeNet5; VGG; ResNet50
2. Improvement one: Resnet50 + Image Attention
3. Improvemnt two: CNN RNN Mixed Model