wilson-99/cifar10-resnet50

CIFAR-10 ResNet-50 Classifier

Model ResNet-50 yang di-fine-tune pada dataset CIFAR-10.

Akurasi

• Test Accuracy: 95.01%

Kelas

airplane, automobile, bird, cat, deer, dog, frog, horse, ship, truck

Cara Pakai

import torch
from torchvision import models, transforms
from PIL import Image

model = models.resnet50(weights=None)
model.fc = torch.nn.Sequential(
    torch.nn.Dropout(0.3),
    torch.nn.Linear(model.fc.in_features, 10)
)
model.load_state_dict(torch.load("model.pth"))
model.eval()