Views
No views yet
torch library in Python:1import torch
2
3# Load the model
4model_path = "path/to/skin_model2.pth"
5model = torch.load(model_path, map_location=torch.device('cpu'))
6model.eval()
7
8# Example usage
9from PIL import Image
10from torchvision import transforms
11
12# Preprocess input image
13transform = transforms.Compose([
14 transforms.Resize((224, 224)),
15 transforms.ToTensor(),
16 transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
17])
18
19image = Image.open("example_input.jpg")
20input_tensor = transform(image).unsqueeze(0)
21
22# Make a prediction
23with torch.no_grad():
24 prediction = model(input_tensor)
25 predicted_class = prediction.argmax(dim=1).item()
26 print(f"Predicted Class: {predicted_class}")