#python #yolov5
Вопрос:
Я пытаюсь использовать пакет yolov5-вывода для обнаружения объектов yolov5. Но это приводит к ошибке во время выполнения.
Это код для обнаружения объектов
from yolov5_inference.infer import DetectObject
import cv2
import numpy as np
class_name = {0: 'head', 1: 'helmet', 2: 'person'}
model = DetectObject(weights='best.pt', classes=class_name)
img_path = 'bus.jpg'
img = cv2.imread(img_path)
print(img.shape)
rs = model.detect(img)
print(rs)
В этом и заключается ошибка.
RuntimeError: The size of tensor a (60) must match the size of tensor b (84) at non-singleton dimension 3
Кто-нибудь сталкивался с этой проблемой?
Комментарии:
1. best.pt является ли модель, которую я обучал, используя ‘ github.com/ultralytics/yolov5 — хранилище.