[发明专利]图像文本方向分类方法、装置、设备及存储介质

权利要求书

1.一种图像文本方向分类方法，其特征在于，包括：

获取已标注文本图像样本集合；

根据所述已标注文本图像样本集合生成文本检测模型；

根据所述已标注文本图像样本集合生成图像文本方向二分类模型；

将输入图像输入所述文本检测模型，得到文本字段外接矩形框集合；

从所述文本字段外接矩形框集合中筛选预设数量的外接矩形框，根据所述外接矩形框的宽高比获取所述输入图像中的文本的初步方向；

所述初步方向为横向或纵向；

根据所述外接矩形框的坐标从所述输入图像中裁剪得到对应的预测文本串图像集合；

将所述预测文本串图像集合中的图像输入所述图像文本方向二分类模型，得到预测文本串图像二分类方向集合；

根据所述预测文本串图像二分类方向集合和所述初步方向获取所述输入图像的文本方向。

2.根据权利要求1所述的图像文本方向分类方法，其特征在于，

所述已标注文本图像样本集合中每张图像文本方向均为0度，且对应标注信息，所述标注信息包括文本字段外接矩形框位置坐标；

所述根据所述已标注文本图像样本集合生成文本检测模型包括：

将所述已标注文本图像样本集合中每张图像分别依次进行90度，180度，270度旋转操作，得到新文本图像集合；

根据所述已标注文本图像样本集合中每张图像的所述标注信息和旋转角度，计算生成得到转换后所述新文本图像集合中每张图像对应的标注信息；

将所述新文本图像集合与所述已标注文本图像样本集合合并，得到文本图像集合；

根据所述文本图像集合生成所述文本检测模型。

3.根据权利要求2所述的图像文本方向分类方法，其特征在于，

所述根据所述已标注文本图像样本集合生成所述图像文本方向二分类模型的步骤，包括：

获取所述已标注文本图像样本集合对应的标注信息，根据所述标注信息筛选得到预设数量宽高比大于预设值的文本标注框集合；

根据所述文本标注框集合从所述文本标注框所属的所述已标注文本图像样本集合中的文本图像对应位置进行裁剪，得到第一训练文本串图像样本集合；

对所述第一训练文本串图像样本集合中每个文本串图像进行逆时针180度旋转操作得到第二训练文本串图像样本集合；

根据所述第一训练文本串图像样本集合和所述第二训练文本串图像样本集合，生成所述图像文本方向二分类模型。

4.根据权利要求3所述的图像文本方向分类方法，其特征在于，所述根据所述第一训练文本串图像样本集合和所述第二训练文本串图像样本集合，生成所述图像文本方向二分类模型的具体步骤包括：

将所述第一训练文本串图像样本集合和所述第二训练文本串图像样本集合中每个文本串图像进行预处理；

将所述第一训练文本串图像样本集合中每张文本串图像的文本方向标记为0度，第二训练文本串图像样本集合中每张文本串图像的文本方向标记为180度；

根据标记后的所述第一训练文本串图像样本集合和所述第二训练文本串图像样本集合，通过预设图像分类算法得到图像文本方向二分类模型。