BlazeFace – это модель машинного обучения, разработанная Google для быстрого определения местоположения и ключевых точек лиц.
Положение лица и ключевые точки лица могут быть получены одновременно. Есть шесть ключевых точек: глаза, нос, уши и рот. Также возможно обнаружение нескольких человек одновременно.
Первоначально модель была предоставлена Google для MediaPipe, но версия, преобразованная в Pytorch, которую может использовать ailia SDK, также представлена в репозитории ниже.
Архитектура
BlazeFace предназначен для очень быстрого получения логических выводов на мобильных графических процессорах. В частности, он работает почти в 2,3 раза быстрее, чем MobileNetV2-SSD.
BlazeFace использует улучшенную сеть на основе MobileNet. Учитывая тот факт, что свертка 3x3 по глубине тензора 56x56x128 занимает 0,07 мс на iPhoneX, в то время как последующая свертка 1x1 от 128 до 128 каналов в 4,3 раза медленнее и составляет 0,3 мс, это показывает, что увеличение размера ядра части по глубине относительно дешево. . Поэтому авторы предлагают заменить свертку по глубине 3x3 на свертку по глубине 5x5, сделав модель более мелкой, чтобы ускорить процесс.
Кроме того, существует фиксированная стоимость диспетчеризации вычислений определенного слоя для шейдеров на графических процессорах. Например, с MobileNetV1 из 4,9 мс времени вывода только 3,9 мс тратится на фактические вычисления шейдера графического процессора. Чтобы уменьшить эту стоимость диспетчеризации для вычисления привязки, авторы приняли альтернативную схему привязки, чтобы уменьшить количество слоев.
использование
Чтобы использовать BlazeFace с ailia SDK, используйте следующий пример.
Вы можете использовать следующую команду для запуска BlazeFace в видеопотоке веб-камеры.
$ python3 blazeface.py -v 0
Приведенную ниже команду можно использовать для запуска BlazeFace на изображении.
$ python3 blazeface.py -i person.jpg
Связанная тема
Топор Инк. разработал ailia SDK, который обеспечивает кросс-платформенный быстрый вывод на основе графического процессора.
ax Inc. предоставляет широкий спектр услуг от консалтинга и создания моделей до разработки приложений и SDK на основе ИИ. Не стесняйтесь связаться с нами для любого запроса.
