animation-classification
animation-classification copied to clipboard
Deep learning model to identify a scene/character from 10 different animated movies
Project: Animated Movie Classification using Deep Learning
Live demo: https://animation-demo.app.render.com/
This is a repo for this detailed blog post. Including Grad-CAM visualization Python module (code and use cases) and notebooks containing training processes and steps. If you want to trace my process, here is the order (for more details you can visit the blog post):
- crop-fix.ipynb
- google-image-download.ipynb
- image-augmentation.ipynb
- resnet-training-with-upsampling.ipynb
- final-training-all-images.ipynb
- live-action-model-check.ipynb
Tools use: Fastai library which built on top of Pytorch 1.0, Matplotlib, scikit-image, google-images-download