Skip to content

Explore deep learning-powered image classification with PyTorch. Achieved 98% accuracy on Natural Images and 95% on Birds Species using AlexNet and EfficientNet-B1. Dive into the code and results!

Notifications You must be signed in to change notification settings

leilibrk/Image-Classification-Deep-Learning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 

Repository files navigation

Image Classification with Deep Learning

This repository contains the code for an image classification project that utilizes deep learning techniques for accurate classification.

Introduction

The project was developed using PyTorch, a widely adopted deep learning framework known for its flexibility and ease of use.

The models were trained on a comprehensive dataset consisting of 100,000 images, and their performance was evaluated on an independent test set comprising 10,000 images.

Achieved Accuracies

Model Dataset Accuracy
AlexNet Natural Images 98%
VGG Natural Images 98%
Your CNN Natural Images 88%
EfficientNet-B1 Birds Species 95%
EfficientNet-B4 Birds Species 85%
AlexNet Birds Species 94%

Usage

To explore this project:

  1. Clone the repository: git clone https://github.com/leilibrk/Image-Classification-Deep-Learning.git
  2. Set up your environment and dependencies. Consider using virtual environments or containers.
  3. Access the Jupyter Notebook files to review the code and results interactively.

Acknowledgments

We extend our appreciation to the PyTorch community for providing an excellent platform for deep learning research and development.

Contributing

Contributions to this repository are welcome! Feel free to fork and create pull requests for any improvements or enhancements.

About

Explore deep learning-powered image classification with PyTorch. Achieved 98% accuracy on Natural Images and 95% on Birds Species using AlexNet and EfficientNet-B1. Dive into the code and results!

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published