You can deploy this model and start earning money today!

marvelmmy/
resnet18-oxford-pet-classifier

Name: resnet18-oxford-pet-classifier
Availability: InStock

Views

No views yet

🐶🐱 Oxford Pets Classifier with ResNet18

This model classifies pet images (cats and dogs) from the Oxford-IIIT Pet Dataset using a fine-tuned ResNet18 architecture.

🧠 Model Details

Architecture: ResNet18
Framework: PyTorch
Trained on: 37 pet breeds
Input size: 224x224 RGB

🗂️ Dataset Oxford-IIIT Pet Dataset: 7,349 images of cats and dogs, labeled with breed.

🏁 Training

Optimizer: Adam
Epochs: 10
Augmentations: RandomRotation, HorizontalFlip
Pretrained: ImageNet weights

📈 Results Achieved ~88% accuracy on validation set.

🧪 Example Inference

python

from PIL import Image
from torchvision import transforms
import torch

transform = transforms.Compose([
    transforms.Resize((224, 224)),
    transforms.ToTensor()
])

img = Image.open("path/to/image.jpg")
input_tensor = transform(img).unsqueeze(0)

model.eval()
with torch.no_grad():
    outputs = model(input_tensor)
    probs = torch.nn.functional.softmax(outputs[0], dim=0)
    predicted_class = torch.argmax(probs).item()

marvelmmy/resnet18-oxford-pet-classifier

marvelmmy/
resnet18-oxford-pet-classifier