Skip to content

This project implements a Convolutional Neural Network (CNN) to classify images from the CIFAR-10 dataset, which consists of 60,000 32x32 color images across 10 classes: airplane, automobile, bird, cat, deer, dog, frog, horse, ship, and truck.

License

Notifications You must be signed in to change notification settings

SunnyBibyan/Image-Classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 

Repository files navigation

CIFAR-10 CNN Classifier

This project demonstrates a CNN-based approach to classify images from the CIFAR-10 dataset using TensorFlow and Keras.

How to Run

  1. Install dependencies:
    pip install -r requirements.txt

Summary

The trained model reached a test accuracy of over 89% with a relatively simple model-structure and only minor overfitting due to different regularization techniques like Dropout, and EarlyStopping. But there is still a lot of optimization potential.

About

This project implements a Convolutional Neural Network (CNN) to classify images from the CIFAR-10 dataset, which consists of 60,000 32x32 color images across 10 classes: airplane, automobile, bird, cat, deer, dog, frog, horse, ship, and truck.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published