Pree46 / Pet-Image-Classifier

This project aims to classify pet pictures, providing accurate breed names based on the chosen pre-trained model. It serves as a handy tool for pet enthusiasts and developers interested in image classification tasks.

Pet Image Classifier

This project provides a simple Python script for image classification using pre-trained models from PyTorch (ResNet, AlexNet, VGG) on ImageNet. The script takes an image and a model name as input, performs image processing, applies the specified pre-trained model, and returns the predicted class label.

Prerequisites

• Python 3.x
• PyTorch
• torchvision
• Pillow (PIL)

Install dependencies using:

pip install torch torchvision pillow

Usage

git clone https://github.com/Pree46/Image-classifier-project
cd Image-Classifier-project
cd home

Models used

• ResNet18
• AlexNet
• VGG16