metadata
license: mit
tags:
- image-classification
- cifar10
- pytorch
- resnet
CIFAR-10 Image Classifier (ResNet-18)
Model klasifikasi gambar CIFAR-10 menggunakan ResNet-18 yang di-fine-tune.
Kelas
plane, car, bird, cat, deer, dog, frog, horse, ship, truck
Akurasi
Test Accuracy: 86.53%
Penggunaan
import torch
model = torch.load('cnn_model_full.pth')
model.eval()