Skip to content

A deep learning project implementing GoogleNet, EfficientNet, and (ViT) for multi-class image classification achieving 97.4% accuracy on 2,484 test images and πŸ† ranking 1st on the Ain Shams AI Kaggle leaderboard.

Notifications You must be signed in to change notification settings

ahany42/CNN-Fruit-Assesment

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

11 Commits
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

🧠 CNN-Based Image Classification Project

This repository contains a deep learning project implementing three powerful CNN architectures layer by layer for multi-class image classification.

πŸš€ Project Overview

  • Models implemented:

    • GoogleNet
    • EfficientNet
    • Vision Transformer (ViT)
  • Dataset: 2,484 images categorized into 7 classes

  • Achieved 97.423% accuracy

  • Ranked 1st (tied) on the Ain Shams AI department Kaggle leaderboard


πŸ“‚ Repository Structure

β”œβ”€β”€ notebooks/
β”‚   └── cnn_classification.ipynb      # Main Jupyter notebook with model training and evaluation
β”œβ”€β”€ testing_scripts/
β”‚   β”œβ”€β”€ test.py                       # Script for testing trained models on new data
β”‚   β”œβ”€β”€ output/                       # Folder where CSV results will be saved
β”‚   β”œβ”€β”€ models/                       # Folder where trained models are saved
β”‚   └── Test/                        # Folder where test photos should be placed
β”œβ”€β”€ README.md  

πŸ† Result

Achieved 97.423% accuracy on 2484 unseen test samples and ranked 1st (tied) on the Ain Shams CS department Kaggle leaderboard

πŸ“‹ How to Use

  1. Open and run the notebook in the notebooks/ folder to train and evaluate models.
  2. Use the testing script in testing_scripts/ to run inference on new images.

About

A deep learning project implementing GoogleNet, EfficientNet, and (ViT) for multi-class image classification achieving 97.4% accuracy on 2,484 test images and πŸ† ranking 1st on the Ain Shams AI Kaggle leaderboard.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 2

  •  
  •