/ants-and-bees-pytorch

Image Classification using PyTorch 1.1.0

Primary LanguagePython

Ants and Bees Classification using PyTorch 1.1.0 (WIP)

This repository is for classifying the ants and bees dataset available from Kaggle here.

The following models will be trained on the dataset and benchmarked

  • VGG16
  • ResNet18
  • AlexNet
  • Ensemble and features and classify using KPCA-MNL