LukaszMalucha / Image-Classification

Flask Blueprint & RESTful application with various image classification models.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Image Classifiers - Various Concepts


1


PROJECT OVERVIEW

Flask RESTful Application hosting three different approaches to image recognition APIs. First, Image Content Classifier, takes image url path as an input. Second model works with .jpg image upload. Third one, Digit Recognition, receives input via js.


Image Content Classification with Tensorflow

Application of Google-made image content classifier built with Tensorflow. Training this kind of model in home conditions would took an eternity, that's why it was downloaded from official website. Model accuracy is amazing.

(Heroku-hosted app works only with small photos with white background due to limited memory, works perfect on local hosts)


2


Cat vs Dog Classifier with Keras & SK

Image classifier trained to distinct between cats and dogs images. Convolutional Neural Network was built with Keras & Tensorflow(GPU).

Kaggle Dataset

CONVOLUTIONAL NEURAL NETWORK CHARACTERISTICS

  1. Image Input Shape - 128,128,3, activation - relu
  2. Three additional Convolutional Layers (batch size - respectively 32,64,128, dropout rate - 0.25,0.2,0.3)
  3. Units in hidden layer - 128
  4. Compiler - optimizer = 'adam', loss = 'binary_crossentropy', metrics = ['accuracy']
  5. Acc - 89% Loss - 25% (approx 30min/epoch on GPU)
  6. CNN Code Location: deep_learning/ConvolutionalNeuralNetwork.py

3


Digit Recognition with Keras & Ajax.js

Hand-Written Digit Recognition based on MNIST Dataset. Convolutional Neural Network was built with Keras & Tensorflow(GPU). Heroku-hosted web application was built with Flask framework, Ajax & FileSaver.

MNIST Dataset


1


Travis CI:

Build Status

Test Files:

/tests

TOOLS, MODULES & TECHNIQUES:

Web Development:

Flask | Flask_Restful | Ajax | FileSaver.js | HTML | CSS | Bootstrap | Materialize | Docker

Python – CNN:

Keras | Tensorflow | Scipy | Numpy | h5py | Scikit-Image

Database Development:

SQLite

Testing

selenium | unittest


CREDITS & INSPIRATIONS

Error 404 template:

Robin Selmer:

https://codepen.io/robinselmer/pen/vJjbOZ


Thank you,

Lukasz Malucha

About

Flask Blueprint & RESTful application with various image classification models.

License:MIT License


Languages

Language:Python 86.6%Language:Gherkin 8.5%Language:HTML 4.9%