Detect whether the image consist of ants or bees using PyTorch
I used RESNET-18 pretrained model to perform classification (Ants OR Bees) on a set of images.
The dataset used is from Kaggle named as "hymenoptera_data".This is the link
The accuracy of network on validation set is 92.15%