/image-classification

Simple image classification models

Primary LanguagePythonMIT LicenseMIT

Image classification

Simple image classification models

Installation

pip install -r requierments.py

Usage

  1. Clone the codes; cd image-classification
  2. Config your parameters in the src/parameters.py module
  3. cd src
  4. Now you can run some of the following commands for running
  5. python LeNet-with-mnist.py (running LeNet over mnist dataset)
  6. python LeNet-with-rps.py (running LeNet over rps dataset)
  7. python MLP-with-mnist.py (running MLP over mnist dataset)

Data-sets

  1. mnist (it will be downloaded by keras)
  2. rps (download rps data ind extract it into a folder by the name data so the data will be as follow ./data/rps)

Models

The following models are used.

LeNet

INPUT => CONV => RELU => POOL => CONV => RELU => POOL => FC => RELU => FC

MLP

Contains two fully connected layers