本文介绍了BlazeFace，一个来自Google Research的轻量级面部检测器，并展示了如何将预训练的TFLite模型转换为PyTorch使用。BlazeFace不仅提供边界框，还预测6个关键点。请注意，这个预训练模型适用于前置摄像头，可能无法检测较小的面部，主要针对自拍场景。提供了转换和推理的Notebooks以及用于测试的图像。
BlazeFace in Python
BlazeFace is a fast, light-weight face detector from Google Research. Read more, Paper on arXiv
A pretrained model is available as part of Google's MediaPipe framework.
Besides a bounding box, BlazeFace also predicts 6 keypoints for face landmarks (2x eyes, 2x ears, nose, mouth).
Because BlazeFace is designed for use on mobile devices, the pretrained model is in TFLite format. However, I wanted to use it from PyTorch and so I converted it.
NOTE: The MediaPipe model is slightly different from the model described in the BlazeFace paper. It uses depthwise convolutions with a 3x3 kernel, not 5x5. And it only uses "single" BlazeBlocks, not "double" ones.
The BlazePaper paper mentions that there are two versions of the model, one for the front-facing camera and one for the back-facing camera. This repo includes only the
最低0.47元/天 解锁文章
扫一扫
1501
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
