Как извлечь только несколько классов из модели Dar knet - PullRequest
0 голосов
/ 29 января 2020

Может быть, этот вопрос странный или я не знаю, что спросить ... Но мне нужно извлечь только несколько классов из целого файла весов модели yolov3.weights.

Мне не нужно получить прогнозы для всех классов, но только немногие. Например: автомобиль, грузовик, автобус, человек и все. Я думаю, что это может ускорить процесс прогнозирования.

В данный момент в моем коде я просто обнаруживаю все на изображении, но отмечаю только объекты OpenCV из списка выше.

Нужно ли мне тренировать мою собственную модель или есть более простой способ? У меня нет доступных наборов данных, и их может быть сложно создать, поскольку я испытываю нехватку времени.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...