Skip to content

MuhammedBuyukkinaci/TensorFlow-Multiclass-Image-Classification-using-CNN-s

Folders and files

NameName
Last commit message
Last commit date
Jan 13, 2019
Nov 19, 2022
Jan 13, 2019
Nov 19, 2022
Nov 19, 2022
Jan 13, 2019
Nov 19, 2022
Nov 19, 2022
Jul 7, 2018
Jan 13, 2019
Nov 19, 2022

Repository files navigation

TensorFlow-Multiclass-Image-Classification-using-CNN-s

This is a multiclass image classification project using Convolutional Neural Networks and PyTorch. If you want to have Tensorflow 1.0 version, take a look at tensorflow1.0 branch.

It is a ready-to-run code.

Folder Tree

folder_tree

Installing Dependencies & Running

python3 -m venv venv
source venv/bin/activate
pip install -r requirements.txt
python main.py

Data

No MNIST or CIFAR-10.

This is a repository containing datasets of 5200 training images of 4 classes and 1267 testing images.No problematic image.

Just extract files from multiclass_datasets.rar.

train_data_bi.npy is containing 5200 training photos with labels.

test_data_bi.npy is containing 1267 testing photos with labels.

Classes are chair & kitchen & knife & saucepan. Classes are equal(1300 glass - 1300 kitchen - 1300 knife- 1300 saucepan) on training data.

Download pure data from here. Warning 962 MB.

Architecture

AlexNet is used as architecture. 5 convolution layers and 3 Fully Connected Layers with 0.5 Dropout Ratio. 60 million Parameters. alt text

Results

Accuracy score reached 87% on CV after just 5 epochs. alt text

folder_tree

Predictions

Predictions for first 64 testing images are below. Titles are the predictions of our Model.

folder_tree