Обычные алгоритмы распознавания лиц обучаются на изображениях без масок, поэтому точность определения положения лица снижается при ношении маски.
FaceMaskDetection может с высокой точностью определять положение лица, даже когда на нем надета маска, и в то же время определять, носит ли лицо маску или нет.
Архитектура
В архитектуре FaceMaskDetection используются YOLOv3 Tiny и MobilenetSSD, модель обучена в ax Inc. В дополнение к обычным изображениям лиц мы добавили в процесс обучения изображения лиц в масках, чтобы определить, носит ли человек маску или нет.
использование
Вы можете увидеть, как использовать FaceMaskDetection, с помощью приведенного ниже примера кода.
Следующая команда может быть использована для определения того, надета ли маска на какое-либо изображение. По умолчанию используется yolov3-tiny.
$ python3 face-mask-detection.py -i input.png -s output.png
Следующая команда запускает модель в видеопотоке веб-камеры.
$ python3 face-mask-detection.py -v 0
Добавьте параметр -a mb2-ssd
для запуска модели с помощью mobilenet-ssd.
$ python3 face-mask-detection.py -v 0 -a mb2-ssd
Топор Инк. разработал ailia SDK, который обеспечивает кросс-платформенный быстрый вывод на основе графического процессора.
ax Inc. предоставляет широкий спектр услуг от консалтинга и создания моделей до разработки приложений и SDK на основе ИИ. Не стесняйтесь связаться с нами для любого запроса.