CNN Image classification of trash of 6 categories in TensorFlow and PyTorch
Trash data categories from Kaggle included:
- Plastic
- Metal
- Paper
- Cardboard
- Glass
- Trash
Implemented a simple CNN network training from scratch on TensorFlow
Implemented Transfer Learning on two ResNet models of different model complexity:
- ResNet18
- ResNext101-32d8 (Experimental Dropout at last FC layer)
Implemented a web interface + acquire new observations for trainning using Flask
Instruction:
- Download link to trained model: https://drive.google.com/file/d/1-2kvwC3w2SdOvROc1RXwUX9bufdM0fC3/view?usp=sharing
- Put downloaded model into /models