[发明专利]一种图像处理方法及装置、存储介质

权利要求书

1.一种图像处理方法，其特征在于，所述方法包括：

在获取到原始图像的情况下，将所述原始图像输入检测对象识别模型中，得到所述原始图像中目标检测对象对应的多个检测框、所述多个检测框的多个得分信息和所述多个检测框中目标检测对象的多个关键点位置信息；

根据所述多个得分信息，从所述多个检测框中筛选出目标检测框；

根据所述目标检测框对应的目标关键点位置信息，确定出原始图像的旋转角度，所述目标关键点位置信息为所述多个关键点位置信息中与所述目标检测检测框对应的关键点位置信息；

根据所述旋转角度旋转所述原始图像并显示旋转后的原始图像。

2.根据权利要求1所述的方法，其特征在于，所述将所述原始图像输入检测对象识别模型中，得到所述原始图像中目标检测对象对应的多个检测框、所述多个检测框的多个得分信息和所述多个检测框中目标检测对象的多个关键点位置信息之前，所述方法还包括：

获取所述样本图像、所述样本图像对应的样本检测框、所述样本检测框的样本得分信息和所述样本检测框对应的样本关键点位置信息；

将所述样本图像输入原始检测对象识别模型中，得到与所述样本图像对应的待训练样本检测框、待训练样本得分信息和待训练样本关键点位置信息；

利用所述待训练样本检测框、所述待训练样本得分信息、所述待训练样本关键点位置信息、所述样本得分信息、所述样本关键点位置信息和所述样本检测框对所述原始检测对象识别模型进行训练，得到所述检测对象识别模型。

3.根据权利要求1所述的方法，其特征在于，所述多个关键点位置信息包括所述目标检测对象的多个左眼角位置信息、多个右眼角位置信息、多个鼻尖位置信息、多个左嘴角位置信息和多个右嘴角位置信息。

4.根据权利要求1所述的方法，其特征在于，所述根据所述目标检测框对应的目标关键点位置信息，确定出原始图像的旋转角度，包括：

将所述目标关键点位置信息输入多维位姿模型中，得到所述目标关键点位置信息对应的多维位置信息；

根据所述多维位置信息，确定所述旋转角度。

5.根据权利要求1所述的方法，其特征在于，所述根据所述多个得分信息，从所述多个检测框中筛选出目标检测框，包括：

根据所述多个得分信息，从所述多个检测框中提取得分最大的第一检测框；

从所述多个检测框中，剔除与所述第一检测框的重叠面积大于预设面积阈值的检测框，得到多个第一候选检测框；

在所述多个第一候选检测框的数量小于或者等于预设数量阈值的情况下，将所述第一检测框作为所述目标检测框。

6.根据权利要求5所述的方法，其特征在于，所述从所述多个检测框中，剔除与所述第一检测框的重叠面积大于预设面积阈值的检测框，得到多个第一候选检测框之后，所述方法还包括：

在所述多个第一候选检测框的数量大于所述预设数量阈值的情况下，继续从所述多个第一候选检测框中提取得分最大的第二检测框；

从所述多个第一候选检测框中，剔除与所述第二检测框的重叠面积大于预设面积阈值的检测框，得到多个第二候选检测框；并将所述多个第二候选检测框的数量与所述预设数量阈值进行对比；

直至所述多个第二候选检测框的数量小于或者等于所述预设数量阈值时，将所述第一检测框和所述第二检测框作为所述目标检测框。

7.一种图像处理装置，其特征在于，所述装置包括：

输入单元，用于在获取到原始图像的情况下，将所述原始图像输入检测对象识别模型中，得到所述原始图像中目标检测对象对应的多个检测框、所述多个检测框的多个得分信息和所述多个检测框中目标检测对象的多个关键点位置信息；

筛选单元，用于根据所述多个得分信息，从所述多个检测框中筛选出目标检测框；

确定单元，用于根据所述目标检测框对应的目标关键点位置信息，确定出原始图像的旋转角度，所述目标关键点位置信息为所述多个关键点位置信息中与所述目标检测检测框对应的关键点位置信息；

旋转单元，用于根据所述旋转角度旋转所述原始图像；

显示单元，用于显示旋转后的原始图像。