- Automate image classification that take leverage from Efficientnet b2 pretrained weights.
- You can have a custom image classifier model with just a few clicks.
- train on google colab with free GPU.
- This is an example model that I've deploy on Hugging Face
- see result model on Youtube
- download this repository : click code > download zip
- create new folder(any name) in your google drive and upload all file in :
- create "raw_data"(exact same name) folders then upload your image folder in :
[recommend] images in each folder shold have more 50 images for better result.
You ready to go open train.ipynb (double click in your drive) and follow the steps in it :)