Garbage_Classifier

A CNN model to predict the type of garbage by training it from a set of 6 different classes.

• Cardboard
• Glass
• Metal
• Paper
• Plastic
• Trash

We've used the architecture of Xception Model along with a learning rate scheduler and custom callbacks to try and prevent over-fitting.

Please refer to my kaggle notebook for detailed comments on the code

The link to my kaggle notebook --> KaggleGarbageClassifier

Thank you!