A binary image classification project using fast.ai and Gradio.
The aim of the classifier is to distinguish between images of Ants and Termites.
This is the link to my Kaggle notebook for the same:
Using the fast.ai library, I have created a binary image classifier involving the following:
- ResNet18 with pretrained weights
- SaveModelCallback( )
- ReduceLROnPlateau( ) Callback
- Image Augmentations:
- Resizing
- Rotations
- Flipping
- Affine transformations
I deployed the model using Gradio, which allows you to build interactive demos to showcase projects.