人脸识别5.1.2insightface人脸检测模型RetinaFacePaddle

知乎RetinaFace&＃43;ArcFace人脸识别测试:https://zhuanlan.zhihu.com/p/352152080
AIStudio:AIStudio仓库
paddle github:https://github.com/GuoQuanhao/RetinaFace-Paddle
pytorch github&＃xff1a;https://github.com/biubug6/Pytorch_Retinaface

&＃xff08;2&＃xff09;进一步添加自监督网络解码器&＃xff08;mesh decoder&＃xff09;分支&＃xff0c;与已有的监督分支并行预测像素级的3D形状的人脸信息。和三维密集对应回归&＃xff08;3D dense correspondence regression&＃xff09;。这种密集的人脸定位为所有不同的尺度提供了精确的人脸位置信息。

1、RetinaFace的人脸检测和人脸对齐代码

人脸检测和人脸对齐部分代码参考&＃xff1a;https://github.com/biubug6/Pytorch_Retinaface&＃xff0c;该代码中也包含人脸识别&＃xff0c;测试过期效果&＃xff0c;不如ArcFace&＃xff0c;感兴趣的可以试下。

2、ArcFace的人脸识别代码

人脸识别代码参考&＃xff1a;https://github.com/TreB1eN/InsightFace_Pytorch。
整合了RetinaFace的人脸检测和人脸对齐代码和ArcFace的人脸识别代码&＃xff0c;使用LFW数据进行测试&＃xff0c;即LFW人脸图片->人脸检测->人脸对齐->人脸识别->测试结果。测试方案如下&＃xff1a;

3、对比

&＃xff08;1&＃xff09;人脸检测中&＃xff0c;对于RetinaFace&＃xff0c;采用代码中的MobileNet0.25-RetinaFace和Resnet50-RetinaFace两种模型&＃xff1b;
&＃xff08;2&＃xff09;人脸对齐中&＃xff0c;代码中的人脸对齐有相似变换&＃xff08;get_similarity_transform_for_cv2&＃xff09;、仿射变换&＃xff08;getAffineTransform&＃xff09;&＃xff1b;为了验证不同变换的效果&＃xff0c;本人也加入了透视变换&＃xff08;getPerspectiveTransform&＃xff09;&＃xff0c;共3种变换进行测试&＃xff1b;
&＃xff08;3&＃xff09;对于ArcFace&＃xff0c;采用代码中的MobileFacenet-ArcFace和IR-SE50-ArcFace两种模型。上述三种情形共组成12种方案。

从图中可以看出&＃xff0c;Resnet50-RetinaFace&＃43;相似变换&＃43;IR-SE50-ArcFace组合取得最优效果&＃xff0c;准确率为99.43%

密集人脸关键点准确性除了人脸框和5点&＃xff0c;Retinaface还输入密集人脸关键点&＃xff0c;并且是自监督训练。
在AFLW2000-3D数据集上评价密集人脸关键点定位
&＃xff08;1&＃xff09;2D投影下的68人脸关键点
&＃xff08;2&＃xff09;所有关键点的3D坐标。平均误差通过人脸框大小归一化。图8a和图8b是当前最好方法的CED曲线。尽管自监督和有监督方法的性能差异还比较大&＃xff0c;但是RetinaFace相比较而言是最好的方法。
特别的&＃xff0c;可以看出&＃xff1a;
&＃xff08;1&＃xff09;五点回归可以避免密集回归分支的训练困难并且可以显著提升密集回归的效果。
&＃xff08;2&＃xff09;使用单级特征来预测密集人脸比使用RoI特征&＃xff08;如网格编码&＃xff09;要困难的多。如图8c所示&＃xff0c;RetinaFace可以很容易的处理人脸姿态变化但是在复杂场景下就比较困难。这表示没有对齐以及过于压缩的特征表示&＃xff08;1x1x256&＃xff09;会妨碍单级框架获取高精度的密集回归输出。尽管如此&＃xff0c;回归分支中投影出来的回归区域依然有助于人类检测结果的提升。

解压数据&＃xff0c;自动创建文件夹

unzip -d Data /home/aistudio/data/data75233/widerface.zip


启动训练

python train.py --network resnet50


代码流程

制作先验框

读取数据、打乱、批量输入格式

构造模型架构

喂入数据&＃xff0c;关键的框四个点损失&＃xff0c;五个关键点损失&＃xff0c;是否是人脸的分类损失

四个坐标点

loss_l &＃61; F.smooth_l1_loss(loc_p, loc_t, reduction&＃61;&＃39;sum&＃39;)


十个坐标点

loss_landm &＃61; F.smooth_l1_loss(landm_p, landm_t, reduction&＃61;&＃39;sum&＃39;)


交叉熵

loss_c &＃61; F.cross_entropy(conf_p, targets_weighted.astype(&＃39;int64&＃39;), reduction&＃61;&＃39;sum&＃39;)

笔记

annotations

详解Python中函数和模块的特殊属性__annotations__
Python是一种动态类型语言&＃xff0c;也是强类型语言。在Python语言中&＃xff0c;使用变量之前不需要声明其类型&＃xff0c;直接赋值即可创建变量&＃xff0c;变量初始类型取决于等号右侧表达式的值的类型。创建之后&＃xff0c;变量的类型可以随时发生变化&＃xff0c;但在任何时刻&＃xff0c;每个变量都有确定的类型。


函数的__annotations__属性只包含形参和返回值的注解&＃xff0c;在Python中&＃xff0c;函数会维护一个特殊属性__annotations__&＃xff0c;这是一个字典&＃xff0c;其中的“键”是被注解的形参名&＃xff0c;“值”为注解的内容。使用时并不要求注解的内容是Python中的类型&＃xff0c;可以是任意内容。

len&＃xff0c;类里面定义&＃xff0c;引用类的时候会触发这个魔法函数

dataset &＃61; WiderFaceDetection(training_dataset, preproc(img_dim, rgb_mean, rgb_std))epoch_size &＃61; math.ceil(len(dataset) / batch_size)


def __len__(self):return len(self.imgs_path)


iter(self)&＃xff0c;for循环的时候触发

for images, labels in batch_iterator:if iteration % epoch_size &＃61;&＃61; 0:if (epoch % 5 &＃61;&＃61; 0 and epoch > 0) or (epoch % 5 &＃61;&＃61; 0 and epoch > cfg[&＃39;decay1&＃39;]):