浅析Mobile net face 2d keypoints alignment 模型

浅析Mobile net face 2d keypoints alignment 模型

模型来自Modelscope的106点人脸关键点-通用领域-2D

模型结构

模型分为一个backbone和两个head, 模型的输入是一张96x96的人脸图片经过归一化预处理后的[1,3,96,96]向量,

大概

模型的backbone采用简单的MobileNet结构, 输入形状是[1,3,96,96], 输出向量不做处理直接输入两个head层
keypoint_head输出需要做后处理,将坐标resize为[106,2],然后缩放为原始图片大小即为最终输出

重点

事实上阿里给出的实例代码并不只有这一个模型, 通过源代码可以知道,实际上是先通过SCRFD人脸检测模型获取人脸,并且还需要旋转正才能作为关键点检测模型的输入