Skip to content

This project implements an image classification model using a Tiny VGG16-inspired Convolutional Neural Network (CNN) trained on a subset of the Food-101 dataset, focusing on just three categories: Pizza, Steak, and Sushi.

Notifications You must be signed in to change notification settings

Coder71005/Costom-Model-Detection

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 

Repository files navigation

Costom-Model-Detection

This project implements an image classification model using a Tiny VGG16-inspired Convolutional Neural Network (CNN) trained on a subset of the Food-101 dataset, focusing on just three categories: Pizza, Steak, and Sushi.

🧠 Model Architecture • A compact, VGG16-inspired CNN (Tiny VGG) • Stack of convolutional layers + ReLU + max pooling • Fully connected (dense) layers for classification

📦 Dataset • Subset of Food-101 dataset with 3 classes: Pizza, Steak, and Sushi • Around 255 training images and 75 testing images per class (as per original split)

⚙️ Libraries Used • Python • PyTorch & Torchvision • Matplotlib, NumPy

About

This project implements an image classification model using a Tiny VGG16-inspired Convolutional Neural Network (CNN) trained on a subset of the Food-101 dataset, focusing on just three categories: Pizza, Steak, and Sushi.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published