You can deploy this model and start earning money today!

Tanishrajput/
Inception-v3

Views

No views yet

Inception-v3 CIFAR-100

This repository contains the Inception-v3 model trained on the CIFAR-100 dataset using PyTorch.
It achieves [83.53]% test accuracy and is ready for inference or fine-tuning.

I hope you find this model useful and easy to integrate into your projects.

Model Description

Architecture: Inception-v3 with auxiliary classifiers
Dataset: CIFAR-100 (100 classes, 60,000 32x32 color images)
Performance: [insert accuracy here]% test accuracy
Framework: PyTorch
Input: 3x32x32 images (RGB)
Output: Class probabilities for 100 classes

🔗 Links

GitHub Repository: CNN Models GitHub Repo
Streamlit App: Link

Usage

Feel free to download and use the model in your own projects. Here's a simple example:

python

import torch
import torch.nn as nn
from torchvision import models, transforms
from PIL import Image

# Load the pretrained model
model = models.inception_v3(aux_logits=False)
model.fc = nn.Linear(model.fc.in_features, 100)  # CIFAR-100 has 100 classes
model.load_state_dict(torch.load("Inception-v3.pt", map_location=torch.device('cpu')))
model.eval()

# Example inference
transform = transforms.Compose([
    transforms.Resize((32, 32)),
    transforms.ToTensor(),
    transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))
])

image = Image.open("example_image.png")
input_tensor = transform(image).unsqueeze(0)  # Add batch dimension

output = model(input_tensor)
predicted_class = output.argmax(1).item()
print("Predicted Class:", predicted_class)

Training Details

Optimizer: Adam
Learning Rate Scheduler: StepLR (step_size=20, gamma=0.1)
Loss Function: CrossEntropyLoss
Weight Decay: 1e-4
Batch Size: 128
Epochs: 30
Data Augmentation: Random horizontal flip, random crop, normalization
Hardware: Trained on 2× NVIDIA T4 GPUs with nn.DataParallel

License

This model is released under the MIT License.

Tanishrajput/Inception-v3