BlazeFace：一种非典型专用检测器

冷不防 2024-04-17 05:58 43阅读 0赞

[MediaPipe][] 是谷歌在 CVPR2019会议上开源的一个感知和增强现实的框架。该框架基于图，可在移动设备、工作站和服务器上跨平台运行，并支持移动 GPU 加速。

[BlazeFace][] 为其中一个重要的组件，因为检测是绝大多数计算机视觉应用的入口。过往检测算法总是从提升性能出发，即便关注轻量级检测器的文章仍不免要在通用数据集上刷一下榜。而谷歌这篇紧贴实际需求，考虑场景限制条件，打造出一款小而专的人脸检测器。[BlazeFace][] 输入设置为  128 × 128 128 \\times 128 128×128，仅检测图中较大的人脸。换个角度看，它也可以在局部区域内检测小脸，即不考虑目标交叉情况下的跟踪。

![face\_detection\_android\_gpu][face_detection_android_gpu]

[BlazeFace][] 是一个轻量且性能出色的人脸检测器，为移动 GPU 推理量身定制。它在旗舰设备上的运行速度为200-1000+ FPS。这种超实时性能使其能够应用于任何增强现实管道，该管道需要准确的面部感兴趣区域作为任务特定模型的输入，例如2D/3D 面部关键点或几何估计，人脸特征或表情分类，以及人脸区域分割。[BlazeFace][] 的贡献包括：

*  一个轻量级的特征提取网络，其灵感来自于 MobileNetV1/V2；
 *  一种对 GPU 友好的 SSD 锚点改良方案；
 *  一种改进的限制分辨率策略替代非极大值抑制。

由于前置和后置相机的焦距以及拍摄物体的典型尺寸不同，[BlazeFace][] 为二者分别建立了模型。除了预测轴对齐的人脸矩形外，[BlazeFace][] 模型还可以生成6个人脸关键点（眼睛中心、耳屏点、嘴部中心和鼻尖）坐标，以便估计面部旋转（滚动角度）。这样可以将旋转的面部矩形传递到视频处理管道的后续特定任务阶段，从而减轻后续处理步骤中显著平移和旋转不变性要求。

[BlazeFace][] 模型构建时考虑下面讨论的四个重要设计因素：

**增大感受野**  
作者注意到深度可分离卷积计算由它们的逐点部分支配。对于  s × s × c s\\times s \\times c s×s×c 输入张量， k × k k \\times k k×k 深度卷积涉及  s 2 c k 2 s^2 c k^2 s2ck2 次乘加运算，而后续  1 × 1 1 \\times 1 1×1 卷积到  d d d 输出通道包含  s 2 c d s^2 c d s2cd 次操作，是深度部分的  d / k 2 d / k^2 d/k2 倍。

在实践中，例如， Apple iPhone X 上的 [Metal Performance Shaders][] 实现，对于 56 × 56 × 128 56\\times 56\\times 128 56×56×128 张量，16位浮点运算中的  3 × 3 3\\times 3 3×3 深度卷积需要0.07 ms，而随后从128到128通道的  1 × 1 1\\times 1 1×1 卷积为0.3 ms，是前者的4.3倍（由于固定成本和内存访问因素，这不如纯算术运算计数差异那么显著）。

这一观察表明，增加深度部分的核尺寸相对成本小。为达到特定的感受野大小，[BlazeFace][] 在 bottleneck 中使用  5 × 5 5\\times 5 5×5 卷积，以增加卷积核大小来减少所需的 bottleneck 数量。如下图所示：

![blocks][]

*  考虑到网络前期 bottleneck 的中间张量通道较少，[BlazeFace][] 设计出与 ResNet 中 BasicBlock 类似的 Single BlazeBlock，以便残差连接在“扩展”（增加）的通道分辨率上执行。
 *  网络深层使用 **double BlazeBlock** ，降低中间通道的同时进一步加速了感受野大小的增长（参见上面的右图）。

**特征提取器**  
[BlazeFace][] 必须考虑较小范围的目标尺度，才能具备较低的计算需求。前置摄像头模型的特征提取器如表 Table 4所示，采用  128 × 128 128\\times 128 128×128 像素的 RGB 输入，由一个2D 卷积以及后跟的5个 Single BlazeBlock 和6个 Double BlazeBlock 组成。最大张量深度（通道分辨率）为96，而最低空间分辨率为  8 × 8 8\\times 8 8×8（而 [SSD][] 将分辨率一直降低到  1 × 1 1\\times 1 1×1）。

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3lpcmFuMTAz_size_16_color_FFFFFF_t_70]  
**锚点方案**

通常的做法是根据目标尺度范围在多个分辨率级别定义锚点。积极的下采样也是优化计算资源的手段。典型的 [SSD][] 模型使用来自  1 × 1 1\\times 1 1×1、 2 × 2 2\\times 2 2×2、 4 × 4 4\\times 4 4×4、 8 × 8 8\\times 8 8×8 和  16 × 16 16\\times 16 16×16 特征图大小的预测。然而，池化金字塔网络（Pooling Pyramid Network，[PPN][]）架构的成功意味着在达到某个特征映射分辨率之后，额外的计算可能是多余的。

相对于 CPU 计算，GPU 的一个关键特性是调度特定层计算的显著固定成本，对于流行的 CPU 定制架构固有的深度低分辨率层来说，这一点变得相对重要。例如，在一个实验中，作者观察到，在 [MobileNetv1][] 的4.9ms 推理时间中，只有3.9ms 用于实际的 GPU 着色器计算。

考虑到这一点，[BlazeFace][] 采用了另一种锚点方案：

*  仅将特征图下采样到  8 × 8 8\\times 8 8×8；
 *  用  8 × 8 8\\times 8 8×8 中的6个锚点替换原来  8 × 8 8\\times 8 8×8、 4 × 4 4\\times 4 4×4 和  2 × 2 2\\times 2 2×2 分辨率中的每个像素对应的2个锚点。
 *  由于人脸长宽比的变化有限，将锚框固定为1：1纵横比。

![anchors][]  
**后处理**  
由于 [BlazeFace][] 的特征提取器没有将分辨率降低到  8 × 8 8\\times 8 8×8 以下，因此与给定目标重叠的锚点数量会随着目标大小的增加而显着增加。在典型的非极大值抑制场景中，只有一个锚点“胜出”并用作最终的算法结果。当将这种模型应用于后续视频帧时，预测倾向于在不同锚框之间波动并且表现出时间抖动（人类可感知的噪声）。

为了最大限度地减少这一现象，[BlazeFace][] 将**抑制**算法替换为**混合**策略，该策略将边界框的回归参数估计为重叠预测之间的加权平均值。它几乎不会给原始的NMS算法带来额外的成本。对于这里的人脸检测任务，该调整使精度提高10%。

> 之前很多人不求甚解，仅根据文中表格创建网络即宣称做出算法的实现。试问检测器没有后处理，仅靠网络能运行吗？细节根本对不上。考虑到谷歌本身就已放出代码，以上不得不令人唏嘘。

## [WeightedNonMaxSuppression][] ##

[WeightedNonMaxSuppression][] 即文中 blending 策略的实现。以加权方式替换原有 NMS 不是首创，但这里的实现和其他方法不完全一样，以分数作为权重进行调和。

[IndexedScores][] 将得分及其索引成对存储。

[Detection][] 可记录检测或跟踪的结果。  
跳过分数低于阈值的检测。  
[Location][] 维护 [LocationData][]。  
[OverlapSimilarity][] 首先从 [Location][] 提取相对框，计算两个 [Location][] 之间的重叠相似性。它假定 [Location][] 中已经有一个相对的框表示，因此不需要帧宽和高度来进一步标准化。

将重叠率大于阈值的检测收集到`candidates`。

IndexedScores remained_indexed_scores;
        remained_indexed_scores.assign(indexed_scores.begin(),
                                       indexed_scores.end());
    
        IndexedScores remained;
        IndexedScores candidates;
        output_detections->clear();
        while (!remained_indexed_scores.empty()) {
        
          const auto& detection = detections[remained_indexed_scores[0].first];
          if (options_.min_score_threshold() > 0 &&
              detection.score(0) < options_.min_score_threshold()) {
        
            break;
          }
    
          remained.clear();
          candidates.clear();
          const Location location(detection.location_data());
          // This includes the first box.
          for (const auto& indexed_score : remained_indexed_scores) {
        
            Location rest_location(detections[indexed_score.first].location_data());
            float similarity =
                OverlapSimilarity(options_.overlap_type(), rest_location, location);
            if (similarity > options_.min_suppression_threshold()) {
        
              candidates.push_back(indexed_score);
            } else {
        
              remained.push_back(indexed_score);
            }
          }

`candidates`中的边框和关键点按得分加权取均值。

auto weighted_detection = detection;
          if (!candidates.empty()) {
        
            const int num_keypoints =
                detection.location_data().relative_keypoints_size();
            std::vector<float> keypoints(num_keypoints * 2);
            float w_xmin = 0.0f;
            float w_ymin = 0.0f;
            float w_xmax = 0.0f;
            float w_ymax = 0.0f;
            float total_score = 0.0f;
            for (const auto& candidate : candidates) {
        
              total_score += candidate.second;
              const auto& location_data =
                  detections[candidate.first].location_data();
              const auto& bbox = location_data.relative_bounding_box();
              w_xmin += bbox.xmin() * candidate.second;
              w_ymin += bbox.ymin() * candidate.second;
              w_xmax += (bbox.xmin() + bbox.width()) * candidate.second;
              w_ymax += (bbox.ymin() + bbox.height()) * candidate.second;
    
              for (int i = 0; i < num_keypoints; ++i) {
        
                keypoints[i * 2] +=
                    location_data.relative_keypoints(i).x() * candidate.second;
                keypoints[i * 2 + 1] +=
                    location_data.relative_keypoints(i).y() * candidate.second;
              }
            }
            auto* weighted_location = weighted_detection.mutable_location_data()
                                          ->mutable_relative_bounding_box();
            weighted_location->set_xmin(w_xmin / total_score);
            weighted_location->set_ymin(w_ymin / total_score);
            weighted_location->set_width((w_xmax / total_score) -
                                         weighted_location->xmin());
            weighted_location->set_height((w_ymax / total_score) -
                                          weighted_location->ymin());
            for (int i = 0; i < num_keypoints; ++i) {
        
              auto* keypoint = weighted_detection.mutable_location_data()
                                   ->mutable_relative_keypoints(i);
              keypoint->set_x(keypoints[i * 2] / total_score);
              keypoint->set_y(keypoints[i * 2 + 1] / total_score);
            }
          }
          remained_indexed_scores = std::move(remained);
          output_detections->push_back(weighted_detection);
        }

## 参考资料： ##

*  [BlazeFace: 亚毫秒级的人脸检测器(含代码)][BlazeFace_]
 *  [BlazeFace Sub-millisecond Neural Face Detection on Mobile GPUs][]
 *  [tkat0/PyTorch\_BlazeFace][tkat0_PyTorch_BlazeFace]
 *  [MediaPipe Read-the-Docs][]
 *  [Combination Features and Models for Human Detection][]
 *  [Object detection via a multi-region & semantic segmentation-aware CNN model][Object detection via a multi-region _ semantic segmentation-aware CNN model]
 *  [论文笔记 《Object detection via a multi-region & semantic segmentation-aware CNN model》 ][_Object detection via a multi-region _ semantic segmentation-aware CNN model_]
 *  [On The Stability of Video Detection and Tracking][]
 *  [200～1000+fps！谷歌公布亚毫秒级人脸检测算法BlazeFace，面向移动GPU][200_1000_fps_BlazeFace_GPU]
 *  [谷歌开源基于 ML 的手部跟踪算法：手机端实时检测，多个手势同时捕捉][ML]
 *  [MediaPipe: A Framework for Perceiving and Augmenting Reality][MediaPipe_ A Framework for Perceiving and Augmenting Reality]

[MediaPipe]: http://github.com/google/mediapipe
[BlazeFace]: https://arxiv.org/abs/1907.05047
[face_detection_android_gpu]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/15/ee1f8e9be59d4e05b840c85e1242afe9.gif
[Metal Performance Shaders]: https://developer.apple.com/documentation/metalperformanceshaders
[blocks]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/15/f6bd6f8c59954ac3aa2e4130d17feecc.png
[SSD]: https://arxiv.org/abs/1512.02325
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3lpcmFuMTAz_size_16_color_FFFFFF_t_70]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/15/c5a67b1994934853b9da1e920228ce00.png
[PPN]: https://arxiv.org/abs/1807.03284
[MobileNetv1]: https://arxiv.org/abs/1704.04861
[anchors]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/15/5ffce1398dd44972af802e5b0e16687d.png
[WeightedNonMaxSuppression]: https://github.com/google/mediapipe/blob/master/mediapipe/calculators/util/non_max_suppression_calculator.cc#L294
[IndexedScores]: https://github.com/google/mediapipe/blob/master/mediapipe/calculators/util/non_max_suppression_calculator.cc#L33
[Detection]: https://github.com/google/mediapipe/blob/master/mediapipe/framework/formats/detection.proto#L27
[Location]: https://github.com/google/mediapipe/blob/master/mediapipe/framework/formats/location.h#L49
[LocationData]: https://github.com/google/mediapipe/blob/master/mediapipe/framework/formats/location_data.proto#L26
[OverlapSimilarity]: https://github.com/google/mediapipe/blob/master/mediapipe/calculators/util/non_max_suppression_calculator.cc#L114
[BlazeFace_]: https://zhuanlan.zhihu.com/p/73741766
[BlazeFace Sub-millisecond Neural Face Detection on Mobile GPUs]: https://sites.google.com/view/perception-cv4arvr/blazeface
[tkat0_PyTorch_BlazeFace]: https://github.com/tkat0/PyTorch_BlazeFace
[MediaPipe Read-the-Docs]: https://mediapipe.readthedocs.io/
[Combination Features and Models for Human Detection]: https://www.cv-foundation.org/openaccess/content_cvpr_2015/papers/Jiang_Combination_Features_and_2015_CVPR_paper.pdf
[Object detection via a multi-region _ semantic segmentation-aware CNN model]: https://www.cv-foundation.org/openaccess/content_iccv_2015/papers/Gidaris_Object_Detection_via_ICCV_2015_paper.pdf
[_Object detection via a multi-region _ semantic segmentation-aware CNN model_]: http://zhangliliang.github.io/2015/05/17/paper-note-ms-cnn/
[On The Stability of Video Detection and Tracking]: https://arxiv.org/pdf/1611.06467.pdf
[200_1000_fps_BlazeFace_GPU]: https://mp.weixin.qq.com/s?__biz=MzUzODkxNzQzMw==&mid=2247483937&idx=1&sn=dd6a70b2e7dac850c188a22c675e607c&chksm=fad12d77cda6a461cb74437a5d811bffac17252cad11615a8b91c67fcba80aaf1c48f4a45118&mpshare=1&scene=1&srcid=0825OYzKofIYCxEAIQmNdD8b&sharer_sharetime=1566716479761&sharer_shareid=e08f1346e20cd3a5de6b23be6624f251&pass_ticket=p3dqkS78I4R%2FS9V%2BtAc7HNgA8FGhyfsxhZYAD9meh%2Bc%3D#rd
[ML]: https://m.leiphone.com/news/201908/WRTFEn6nK30Hu8Nr.htm%E2%80%A6%E2%80%A6l
[MediaPipe_ A Framework for Perceiving and Augmenting Reality]: https://sites.google.com/view/perception-cv4arvr/mediapipe