CODEWITHSUNDEEP
pythonmachine-learningpytorchresnetfast-ai
Hassan
Hassan

Reputation: 31

How to Load Fastai model and predict it on single image

I have trained a fastai model using Kaggle notebook, it has saved the model but how to load the model is the problem, i have tried different methods like the method given below. Even it does load the model it doesn't have any predict function only thing I can see is model.eval(). The second problem is when the model was trained on google collab it didn't even get the single image, I did try to convert the image to NumPy way and another way but both didn't work out.

I am attaching the kaggle link of model training, the saved model and the test images in last after this code

#Code for Loading model
from fastai import * 
from fastai.vision import *
import torch
loc = torch.load('/content/gdrive/MyDrive/Data Exports/35k data/stage-1.pth')
body = create_body(models.resnet18, True, None)
data_classes = 4
nf = callbacks.hooks.num_features_model(body) * 2
head = create_head(nf, data_classes, None, ps=0.5, bn_final=False)
model = nn.Sequential(body, head)

Kaggle Model

Test Images From Kaggle Dataset

Saved Model

Upvotes: 3

Views: 3093

Answers (1)

Tinu
Tinu

Reputation: 2513

How to load pytorch models:

loc = torch.load('/content/gdrive/MyDrive/Data Exports/35k data/stage-1.pth')
model = ... # build your model
model.load_state_dict(loc)
model.eval()

Now you should be able to simply use the forward pass to generate your predictions:

input = ... # your input image
pred = model(input) # your class predictions

Don't forget to convert your inputs to torch tensors first, you might want to use a DataLoader for ease of use.

Upvotes: 2

Related Questions