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

权利要求书

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

获取目标图像，所述目标图像中包含一个或多个目标对象；

根据所述目标图像，利用预先训练完成的图像分割模型得到目标检测框以及与所述目标检测框对应的图像分割结果图；

根据所述目标检测框和所述图像分割结果图，对所述目标图像进行裁剪，得到与所述目标对象对应的目标区域；

利用光学字符识别算法确定所述目标区域中所述目标对象的内容，得到识别结果。

2.根据权利要求1所述的方法，其特征在于，所述根据所述目标检测框和所述图像分割结果图，对所述目标图像进行裁剪，得到与所述目标对象对应的目标区域，包括：

根据所述目标检测框对所述目标图像进行裁剪，得到第一目标图像；

根据所述图像分割结果图，确定所述图像分割结果图中分割出来的所述目标对象的最大连通区域；

根据所述最大连通区域的轮廓的像素点，得到所述轮廓的最小倾斜矩阵；

根据所述最小倾斜矩阵的倾斜角度将所述第一目标图像转正；

根据所述最小倾斜矩阵的宽和高对所述转正后的第一目标图像进行裁剪，得到与所述目标对象对应的目标区域。

3.根据权利要求1所述的方法，其特征在于，在所述根据所述目标图像，利用预先训练完成的图像分割模型得到与所述目标对象对应的目标检测框以及与所述目标检测框对应的图像分割结果图之前，还包括：

将所述目标图像输入到预先训练完成的角度分类模型中，得到所述目标图像的角度分类结果，根据所述角度分类结果将所述目标图像进行旋转；

根据所述旋转后的目标图像，利用预先训练完成的图像分割模型得到目标检测框以及与所述目标检测框对应的图像分割结果图。

4.根据权利要求1所述的方法，其特征在于，所述图像分割模型包括目标检测层和图像分割层，所述目标检测层用于对所述目标图像进行特征提取和目标检测，得到目标特征信息和目标检测框，所述图像分割层用于根据所述目标特征信息以及所述目标检测框，得到所述图像分割结果图。

5.根据权利要求4所述的方法，其特征在于，所述图像分割层用于根据所述目标特征信息以及所述目标检测框，得到所述图像分割结果图，包括：

所述图像分割层用于确定所述目标特征信息中与所述目标检测框对应的第一目标特征信息，计算所述第一目标特征信息中的每个像素点的概率值，根据所述每个像素点的概率值，得到所述图像分割结果图。

6.根据权利要求1所述的方法，其特征在于，所述获取目标图像之前，所述方法还包括生成图像分割模型，包括：

获取第一样本图像以及所述第一样本图像中包含目标对象的第一目标检测框；

根据所述第一样本图像和所述第一目标检测框，对所述图像分割模型中的目标检测层进行模型训练，得到第一目标检测层；

获取第二样本图像以及所述第二样本图像中包含目标对象的第二目标分割图；

根据所述第二样本图像和所述第二目标分割图，对所述图像分割模型中的第一目标检测层和图像分割层进行模型训练。

7.根据权利要求6所述的方法，其特征在于，所述根据所述第一样本图像和所述第一目标检测框，对所述图像分割模型中的目标检测层进行模型训练，得到第一目标检测层，包括：

将所述第一样本图像输入到所述图像分割模型中的目标检测层，得到第一预测目标检测框；

根据所述第一预测目标检测框和所述第一目标检测框，确定第一损失函数；

根据所述第一损失函数，更新所述目标检测层的参数，得到第一目标检测层。