下面是我尝试运行的代码。fasterrcnn_foodtracker.pth是我正在尝试使用 PyTorch 加载的已经训练好的模型。
import torch
import torchvision
import cv2
model = torchvision.models.detection.fasterrcnn_resnet50_fpn(pretrained=True)
path = '/home/amir/PycharmProjects/Food-Recognition/fasterrcnn_foodtracker.pth'
model.load_state_dict(torch.load(path, map_location=torch.device('cpu')), strict=False)
model.eval()
img = cv2.imread('twodishes.jpg')
prediction = model([img])
print(prediction)
出现大小不匹配的运行时错误。
RuntimeError: Error(s) in loading state_dict for FasterRCNN:
size mismatch for roi_heads.box_predictor.cls_score.weight: copying a param with shape torch.Size([100, 1024]) from checkpoint, the shape in current model is torch.Size([91, 1024]).
size mismatch for roi_heads.box_predictor.cls_score.bias: copying a param with shape torch.Size([100]) from checkpoint, the shape in current model is torch.Size([91]).
size mismatch for roi_heads.box_predictor.bbox_pred.weight: …