单据文本检测模型的训练方法及单据文本检测方法
摘要:
本发明提供一种单据文本检测模型的训练方法及单据文本检测方法。该训练方法包括:获取训练集图像,其中,训练集图像包括单据图像以及单据图像中文本区域对应的位置标签;对每张单据图像进行特征提取,得到第一特征图;再对第一特征图进行特征融合,得到第二特征图;基于第二特征图得到概率图和阈值图,并对概率图和阈值图进行可微分二值化处理,得到二值图以及单据图像的文本区域位置坐标;基于位置标签、文本区域位置坐标以及预设损失函数对各个参数进行迭代更新,当预设损失函数收敛时,获得训练完成的目标单据文本检测模型。本发明能够有效提升模型的训练精度,进而进一步提高单据文本的检测精度。
0/0