Проблема с классификацией с использованием Yolo Darknet - PullRequest
0 голосов
/ 30 октября 2019

Я обучил свой пользовательский детектор объектов, используя yolo darknet, в 3 классах:

 - human (full body)
 - high visibility jacket (HVJ)
 - safety helmet

Он отлично работает! Но мне нужно добавить функции:

 - if in the frame an human wears nothing (not HVJ/helmet), it is tracked with red box
 - if in the frame an human wears an HVJ, it is tracked with yellow box 
 - if in the frame an human wears an helment, it is tracked with yellow box
 - if in the frame an human wears an helmet and an HVJ, it is tracked with green box

Как я могу это реализовать? В настоящее время с моей обученной моделью не могу этого сделать. Я думаю, что мог бы создать новый набор данных со следующими классами и снова обучить модель классам:

- human with nothing 
- human with only HVJ
- human with only helmet
- human with HSJ and helmet

Но я не уверен, что этот метод может работать. Можете ли вы дать мне несколько предложений? Если это может быть полезно узнать для режима тестирования, я использую скрипт в python / OpenCV.

...