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

# Load model
checkpoint = torch.load('model.pth')
model = models.resnet18()
model.fc = torch.nn.Linear(512, 7)
model.load_state_dict(checkpoint['state_dict'])
model.eval()

# Preprocess image
transform = transforms.Compose([
    transforms.Resize(256),
    transforms.CenterCrop(224),
    transforms.ToTensor(),
    transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
])

# Predict
img = Image.open('maize_leaf.jpg').convert('RGB')
img_tensor = transform(img).unsqueeze(0)

with torch.no_grad():
    output = model(img_tensor)
    probabilities = torch.nn.functional.softmax(output[0], dim=0)
    pred_class = output.argmax(1).item()

print(f"Prediction: {checkpoint['class_names'][pred_class]}")
print(f"Confidence: {probabilities[pred_class]:.2%}")

Model Details

Architecture: ResNet-18 (torchvision)
Input: 224×224 RGB images
Training: Transfer learning from ImageNet
Framework: PyTorch

albertosei/agricom-maize-new-resnet18

Agricom Ghana Maize Disease Classifier

Classes

Usage

Model Details

albertosei/
agricom-maize-new-resnet18