项目作者: urastogi885

项目描述 :
Monkey species classification on the Kaggle dataset using CNNs
高级语言: Python
项目地址: git://github.com/urastogi885/monkey-classifier.git
创建时间: 2021-01-21T17:33:56Z
项目社区:https://github.com/urastogi885/monkey-classifier

开源协议:MIT License

下载


Monkey Classifier

License

Overview

In this project, we perform monkey species classification using convolutional neural networks (CNNs). We use the 10
Monkey Species Dataset
. I have
used my digit recognition CNN as a base for
the custom CNN. It has been updated to include 2 more 2D convolutional layers making it a 4 convolutional layer network.
The base network for transfer learning is VGG16. The VGG model has been appended with 2 fully-connected layers and a
dropout layer. Apart from that, the last 5 layers of the VGG model were made trainable to get better generalization.

Dependencies

  • Python 3.8
  • Pandas 1.0.4
  • Keras 2.4.3
  • Tensorflow 2.4.0

Stick to these versions of dependencies, especially Keras and Tensorflow, if you want to directly use the scripts.

Todo

  • Switch to PyTorch

Run

  • Run the custom 4-layer network and transfer learning network using Python.
  • Note if you do not have GPU properly setup with Tensorflow, the network would take hours to train. With GPU, training
    would take around 20 minutes.
  • Open terminal, head to the project directory, and run the following python scripts:
    1. cd <project directory/Code>
    2. python my_cnn.py
    3. python transfer_learning.py
  • Note that you might have to use your alias for Python-3