Skip to content

A CNN for CIFAR-10 image classification using Keras, featuring data augmentation, BatchNormalization, Dropout, early stopping, and learning rate scheduling. Includes model performance visualization and misclassified image analysis. Built with TensorFlow and Keras.

License

Notifications You must be signed in to change notification settings

edwardmagongo/Model-cifar10-image-classification-edwardmagongo

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CIFAR-10 Image Classification

Project Goal:

The goal of this project is to classify images in the CIFAR-10 dataset using a deep Convolutional Neural Network (CNN). The project includes advanced features such as data augmentation, batch normalization, dropout, and learning rate scheduling to enhance the model's accuracy and prevent overfitting.

Important Files:

  • model.py: The main Python script that contains the CNN model architecture and training logic.
  • requirements.txt: The required Python libraries for running the project.
  • README.md: Documentation for understanding and running the project.

To Use This Project:

  1. Clone the repository:
    git clone https://github.com/edwardmagongo/model-cifar10-image-classification-edwardmagongo.git

About

A CNN for CIFAR-10 image classification using Keras, featuring data augmentation, BatchNormalization, Dropout, early stopping, and learning rate scheduling. Includes model performance visualization and misclassified image analysis. Built with TensorFlow and Keras.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages