[发明专利]一种证件照头发颜色矫正方法、装置及介质

权利要求书

1.一种证件照头发颜色矫正方法，其包括如下步骤：

S1，使用头发分割卷积神经网络对证件照进行头发分割，获取第一分割图像，所述第一分割图像为仅包含头发区域的图像；所述头发分割卷积神经网络是使用MobileNetV2替换SegNet网络中的encoder网络结构；

S2，使用头发颜色合格判断模型判断第一分割图像的头发颜色是否合格；

S3，在所述头发颜色不合格时，对所述证件照中的头发颜色进行染色以获取矫正后的证件照；

对所述证件照的头发颜色进行染色包括如下步骤：

S31，对所述证件照的头发区域进行去色处理，将头发区域颜色的色相和饱和度信息去掉，只保留亮度信息；

S32，对去掉色相和饱和度信息的头发区域进行亮度调整；

所述步骤S32，在亮度调整时，先统计头发区域颜色亮色像素或者是暗色像素的占比，当所述亮色像素占比大于第一预设值时，对所述亮度进行调节；直至所述亮色像素占比小于第一预设值，结束亮度调整；

所述亮度调整公式如下：

其中，为像素的生成的像素值，P为像素的原像素值，B为明亮系数取值范围为[-1,1]，k为约束系数。

2.根据权利要求1所述的方法，所述步骤S2具体为：

S21、获取所述第一分割图像中头发区域总像素N；

S22、创建长宽为（n，n）的三通道图像O，其中；

S23、将所述头发区域的像素按照顺序填充到所述三通道图像O中；

S24、将所述三通道图像O输入头发颜色合格判断模型判断所述头发的颜色是否合格。

3.一种证件照拍照机，所述拍照机包括如下单元：

头发分割单元，用于使用头发分割卷积神经网络对证件照进行头发分割，获取第一分割图像，所述第一分割图像为仅包含头发区域的图像；所述头发分割卷积神经网络是使用MobileNetV2替换SegNet网络中的encoder网络结构；

头发颜色合格判断单元，用于使用头发颜色合格判断模型判断第一分割图像的头发颜色是否合格；

头发颜色矫正单元，用于在所述头发颜色不合格时，对所述证件照中的头发颜色进行染色以获取矫正后的证件照；

所述证件照还包括如下单元：

去色单元，用于对所述证件照的头发区域进行去色处理，将头发区域颜色的色相和饱和度信息去掉，只保留亮度信息；

亮度调整单元，对去掉色相和饱和度信息的头发区域进行亮度调整；

所述亮度调整单元，在亮度调整时，先统计头发区域颜色亮色像素或者是暗色像素的占比，当所述亮色像素占比大于第一预设值时，对所述亮度进行调节；直至所述亮色 像素占比小于第一预设值，结束亮度调整；

所述亮度调整公式如下：

其中，为像素的生成的像素值，P为像素的原像素值，B为明亮系数取值范围为[-1,1]，k为约束系数。

4.一种非易失性存储介质，所述介质上存储有指令，所述指令在被执行时，用以实现如权利要求1-2中任一项所述的证件照头发颜色矫正方法。