Я пытаюсь использовать модель YOLO v2-tiny (модель, обученную на наборе данных COCO) для выполнения вывода изображений (используя набор инструментов OpenVINO).
Я получаю вывод как 1x13x13x425
и узнал, как его выходные данные закодированы / упакованы, и, кажется, получили масштабированные блоки, но у меня возникают проблемы с получением правильных прогнозов. Я имею в виду, что detected не определен правильно, et c. Я предварительно обрабатываю кадр следующим образом:
frame_copy = cv2.resize(frame, (w,h))
frame = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR)
frame = frame_copy.transpose((2,0,1))
frame = frame.reshape((b,c,h,w))
np_img = np.copy(frame)
np_img = np_img.astype('float32')
np_img /= 255.0
Это один проблеск:
Могу ли я получить некоторую помощь? Возможно, какое-то руководство, чтобы следовать или помочь, что может go неправильно ...