OpenCV обнаружение сложного объекта - PullRequest
1 голос
/ 19 октября 2019

Я хотел бы распознать определенный объект на изображении и определить соответствующие координаты. Для этого я использую OpenCV под Python. Объектом для поиска является цилиндрическая светодиодная лампа (см. Рисунок). Лампа имеет определенную яркость, розовый оттенок, и по цилиндрической форме лампа на двухмерном изображении распознается как треугольник.

Lamp

Лампадолжен быть распознан на снимке комнаты. Как вы можете видеть на картинке, лампа, к сожалению, освещена неравномерно. Моя попытка извлечь лампу через порог HSV была не очень успешной. Внутренняя часть лампы белого цвета, а границы недостаточно ограничены.

Есть ли лучший способ надежного обнаружения лампы?

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

Лампа должна быть обнаружена впоследствии в реальном времени на записанном видео в реальном времени. Лампа также движется по комнате.

Я был бы очень признателен за совет.

РЕДАКТИРОВАТЬ: Я хотел бы использовать Raspberry Pi с камерой. Соответственно, алгоритм должен быть ресурсосберегающим.

1 Ответ

0 голосов
/ 19 октября 2019

Я бы порекомендовал использовать легкую предварительно обученную модель обнаружения объектов, такую ​​как YOLO или SSD, с чем-то вроде mobilenet backend .

Существует множество руководств, показывающих, как люди достигли разумного (5-10 кадров / с) с использованием этих моделей на Raspberry Pis

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