Расшифровка Yolo вывода - PullRequest
       41

Расшифровка Yolo вывода

0 голосов
/ 27 апреля 2020

Я пытаюсь использовать модель 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

Это один проблеск: enter image description here

Могу ли я получить некоторую помощь? Возможно, какое-то руководство, чтобы следовать или помочь, что может go неправильно ...

...