Views
No views yet
| Model | Accuracy | F1-Score |
|---|---|---|
| ResNet-50 | 100.00% | 100.00% |
| EfficientNet-B3 | 98.89% | 98.89% |
| MobileNet-V3 | 98.52% | 98.52% |
| Custom CNN | 85.19% | 84.95% |
1import torch
2from torchvision.models import resnet50
3
4model = resnet50()
5model.fc = torch.nn.Sequential(
6 torch.nn.Dropout(0.4),
7 torch.nn.Linear(2048, 6)
8)
9model.load_state_dict(
10 torch.load("best_ResNet-50.pth", map_location="cpu")
11)
12model.eval()