Собственная модель yolov5 /python - ошибка TypeError: 'dict' object is not callable

Я обучил собственную модель на архитектуре yolov5 ('nissan.pt')и пытаюсь её запустить из кода python Использую пример отсюда: https://github.com/ultralytics/yolov5 Поменял только одну команду. Было: model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True). Стало: model = torch.load('nissan.pt', map_location='cpu')

Python выдаёт ошибку: print(model(imgs)) TypeError: 'dict' object is not callable.

Как исправить? Заранее спасибо.

import torch
from PIL import Image
import cv2 as cv

# Model
model = torch.load('nissan.pt', map_location='cpu')

print(model)

# Images
img1 = cv.imread('car1.jpg')
img2 = cv.imread('ca2.jpg')
imgs = [img1, img2]

# Inference
print(model(imgs))

Ответы (0 шт):