[发明专利]一种图像检测方法、装置、电子设备及存储介质

权利要求书

1.一种图像检测方法，其特征在于，包括：

获取待检测图像；

利用联合模型对所述待检测图像进行人脸识别及人脸角度检测，获得所述联合模型输出的检测结果，所述检测结果包括所述待检测图像是否含有人脸以及人脸角度；所述联合模型由多张训练图像及每张训练图像对应的标签训练获得，所述标签包括人脸标签和角度标签。

2.根据权利要求1所述的方法，其特征在于，所述标签还包括人脸标注框坐标，所述方法，还包括：

获取多张训练图像，所述多张训练图像包括人脸图像和非人脸图像，并分别对每张训练图像进行标注，获得每张训练图像对应的人脸标签、角度标签和人脸标注框坐标；

将所述训练图像作为输入，所述训练图像对应的人脸标签、角度标签和人脸标注框坐标作为输出对神经网络进行训练，获得所述联合模型。

3.根据权利要求2所述的方法，其特征在于，所述根据所述待检测图像利用联合模型进行人脸识别及人脸角度检测，获得所述联合模型输出的检测结果，包括：

将所述待检测图像输入所述联合模型中进行人脸识别和人脸角度检测，获得所述待检测图像中是否包含人脸、人脸角度以及人脸标注框坐标。

4.根据权利要求2所述的方法，其特征在于，所述根据所述待检测图像利用联合模型进行人脸识别及人脸角度检测，获得所述联合模型输出的检测结果，包括：

判断所述待检测图像中是否包含人脸；

若包含，则获取该人脸对应的人脸角度以及人脸标注框坐标，所述人脸标注框坐标包括所述人脸标注框的左上角坐标和右下角坐标。

5.根据权利要求3或4所述的方法，其特征在于，所述检测结果还包括所述待检测图像中的正脸图像；相应地，所述根据所述待检测图像利用联合模型进行人脸识别及人脸角度检测，获得所述联合模型输出的检测结果，还包括：

根据所述人脸角度和所述人脸标注框坐标进行图像旋转，获得对应的正脸图像。

6.根据权利要求3或4所述的方法，其特征在于，所述检测结果还包括所述待检测图像中的正脸标注框坐标；相应地，所述根据所述待检测图像利用联合模型进行人脸识别及人脸角度检测，获得所述联合模型输出的检测结果，还包括：

根据所述人脸角度和所述人脸标注框坐标进行坐标映射，获得对应的正脸标注框坐标。

7.根据权利要求2所述的方法，其特征在于，所述将所述训练图像作为输入，所述训练图像对应的人脸标签、角度标签和人脸标注框坐标作为输出对神经网络进行训练，获得所述联合模型，包括：

将一张训练图像输入所述神经网络，所述神经网络输出该训练图像对应的人脸预测值、角度预测值和人脸标注框预测坐标；

根据所述训练图像对应的人脸预测值和人脸标签计算获得人脸标签损失值，根据角度预测值和角度标签计算获得角度标签损失值，根据人脸标注框预测坐标和人脸标注框坐标计算获得人脸标注框坐标损失值；

根据所述人脸标签损失值、所述角度标签损失值和所述人脸标注框坐标损失值计算所述训练图像对应的总损失值；

基于所述总损失值对所述神经网络中的参数进行优化，获得优化后神经网络；

获取新的训练图像继续训练所述优化后神经网络，直至满足训练结束条件，得到所述联合模型。

8.一种图像检测装置，其特征在于，包括：

获取模块，用于获取待检测图像；

检测模块，用于利用联合模型对所述待检测图像进行人脸识别及人脸角度检测，获得所述联合模型输出的检测结果，所述检测结果包括所述待检测图像是否含有人脸以及人脸角度；所述联合模型由多张训练图像及每张训练图像对应的标签训练获得，所述标签包括人脸标签和角度标签。