извлечение списка обнаружений из YOLOv3

Question

Я работаю над приложением для обнаружения объектов с использованием YOLOv3. Я переобучил модель для обнаружения двух классов, и до сих пор это было сделано успешно благодаря создателям Repo. Теперь мне нужно развернуть модель на jetson nano kit, и она должна работать непрерывно и извлекать список обнаружений, в котором указаны классы, координаты достоверности и ограничивающие прямоугольники, затем я должен выполнить некоторую обработку ранее упомянутых параметров, а затем отправить ее с использованием любого протокола связи для в любом случае, другой ECU Как извлечь упомянутые параметры из YOLOv3 и заставить этот код работать непрерывно

Answer 1

Если вы используете сценарий Python для обнаружения YOLOv3, вы встретите эту часть в сценарии:

detections = darknet.detect_image(netMain, metaMain, darknet_image, thresh=0.25)
image = cvDrawBoxes(detections, frame_resized)
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)

Здесь обнаружения должны быть словарем или массивом некоторого вида (он различается в зависимости от реализации). Он должен содержать детектированные_классы , number_of_detections , trust_scores и bounding_boxes . Вы можете изменить код, чтобы извлечь все, что вы хотите. Код выше от этого репо. Я также использовал YOLOv3 для непрерывного обнаружения изображений. Предполагалось сказать, содержит ли изображение человека или нет.

Примечание: Если я правильно помню, в C ++ сценарии YOLOv3 должны быть похожей частью, как я упоминал выше.