Skip to content

Latest commit

 

History

History
14 lines (7 loc) · 354 Bytes

File metadata and controls

14 lines (7 loc) · 354 Bytes

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