Image-Classification-by-Keras-and-Tensorflow

Image Classification using Keras as well as Tensorflow. Image Classification is a Machine Learning module that trains itself from an existing dataset of multiclass images and develops a model for future prediction of similar images not encountered during training. Developed using Convolutional Neural Network (CNN).

This repository contains implementation for multiclass image classification using Keras as well as Tensorflow. These two codes have no interdependecy on each other.

Convolutional Neural Network (CNN)

To be updated soon.

Modules Required

You can download the modules in the respective requirements.txt for each implementation.

Implementation using Keras

sys
os
time
keras==2.2.0
numpy==1.14.5

Implementation using Tensorflow

os
time
cv2
glob
dataset==1.1.0
tensorflow==1.15.0
numpy==1.14.5
sklearn==0.19.1

Steps

1. Install the modules required based on the type of implementation.
2. Download the dataset you want to train and predict your system with. For sample data, you can download the 101_ObjectCategories (131Mbytes).
3. Construct the folder sub-structure required.

Keras

Tensorflow