/bee_classifier_using_cnn

Code for The Metis Challenge: Naive Bees Classifier using CNN.

Primary LanguagePython

Bee Classification using Convolutional Neural Network

A system for classifying bees (honey bee or a bumble bee).

Summary

This is done to compete in The Metis Challenge: Naive Bees Classifier competition.

  • The process_data reads the images, scales it, creates synthetic dataset to make the classifier work well by creating reflected and translated versions of images. It then takes care of class imbalance by undersampling this enhanced dataset. All this processes data is then stored in numpy arrays.
  • The CNN architecture is built/run on Theano making use of abstraction libraries like Lasagne.
  • I've experimented with various architectures (will upload the configurations soon!)

Code

The following modules are required to run the system:

  • Python 2.7
  • NumPy
  • Theano
  • Lasagne
  • Pandas
  • scikit-learn
  • Matplotlib