人脸检测——Faster R-CNN

Posted 2023-03-02 shuzfan

本次介绍人脸检测方法Faster R-CNN：

《2016 Arxiv: Face Detection with the Faster R-CNN》.

上面这篇文章，是对Faster R-CNN的人脸检测实现，原始的Faster R-CNN实现的是多目标检测，即下面这篇文章：

《2015 CVPR: Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks》.

核心导读：

RPN(Region Proposal Networks) + Fast R-CNN

RPN负责找到可能的目标窗口，R-CNN负责进一步判断目标。

因为讲解Faster R-CNN的文章已经很多了，所以我这里就快速的切入几个要点。

—————————— RPN ——————————

RPN负责从一副输入图像中选出一些候选目标窗口，它的作用和古老的“Sliding Window”(滑动窗口)类似，但后者通常会在一幅图像上产生数以万计的窗口。

下面给出RPN的示意图：

输入： 一幅图像，尺寸\\(M\\times600\\), 即resize使得最短边为600.

中间输出： conv feature map，大小\\(m\\times n\\)。在该feature map上以\\(3\\times3\\)滑窗，为了检测不同形状的窗口，这里每一个滑动窗口代表了可能的k=9种实际窗口(分别是3种不同尺度和3种不同宽高比)。每一个实际窗口，我们都称之为anchor。

最终输出： 一是：classifier score map,大小为\\(m\\times\\ n\\times 2\\times k\\),表示每一个窗口是或是不是目标的概率；
二是：regression map, 大小为\\(m\\times\\ n\\times 4\\times k\\),表示每一个窗口所回归出来的四个坐标\\([x,y,w,h]\\)

Test的时候，我们可以根据score map以一定阈值选出候选窗口，然后以regression map来修正这些候选框。

显然，RPN的Loss包含了两个部分，分别是分类和回归：

其中，\\(p_i\\) 是第 \\(i\\) 个anchor属于目标的概率。

—————————— Faster R-CNN ——————————

现在给出整体的结构图：

由上图可以看出，RPN负责提供候选窗口，R-CNN负责对这些窗口做进一步判断，一共是先后的两部分。
而且，两部分共用前面的所有的层，这样就通过参数共享大大减少了模型大小，从而提高了速度。
最后做目标确认的时候，使用了ROI-Pooling方法，即只pooling目标区域。

—————————— 训练策略 ——————————

另外还需要注意的就是训练策略了。

论文最终采用了“Alternating training”的训练策略，分为4阶段：

（1）训练RPN;

（2） 利用RPN产生的候选窗口来训练R-CNN;

（3）在(2)的基础上训练再次RPN，只不过保持共享层不训练；

（4）在(3)的基础上训练R-CNN分支，同时保持其它层不变。