[发明专利]检测模型训练方法、单字检测方法、装置、设备及介质

权利要求书

1.一种文本单字检测模型训练方法，其特征在于，包括：

获取样本图像数据，其中，所述样本图像数据包括标注单字位置的样本位置数据以及包括标注文字数量的样本数量数据；

基于所述样本位置数据生成基础训练样本数据，基于所述样本数量数据生成弱监督训练样本数据；

使用所述基础训练样本数据和所述弱监督训练样本数据对预先构建的文本单字检测模型进行训练，得到训练好的文本单字检测模型。

2.根据权利要求1所述的方法，其特征在于，所述使用所述基础训练样本数据和所述弱监督训练样本数据对预先构建的文本单字检测模型进行训练，得到训练好的文本单字检测模型，包括：

使用所述基础训练样本数据对预先构建的文本单字检测模型进行训练，以基础损失值收敛为目标，得到初始文本单字检测模型；

使用所述弱监督训练样本数据对所述初始文本单字检测模型进行训练，以弱监督损失值收敛为目标，得到迭代文本单字检测模型；

依次使用所述基础训练样本数据和所述弱监督训练样本数据对待训练文本单字检测模型进行迭代训练，得到所述迭代文本单字检测模型，并将每次训练得到的所述迭代文本单子检测模型作为下一次训练的待训练文本单字检测模型，以所述基础损失值和所述弱监督损失值均收敛为目标，得到训练好的文本单字检测模型。

3.根据权利要求2所述的方法，其特征在于，所述使用所述弱监督训练样本数据对所述初始文本单字检测模型进行训练，以弱监督损失值收敛为目标，得到迭代文本单字检测模型，包括：

基于所述初始文本单字检测模型和辅助单字检测算法生成所述弱监督训练样本数据对应的伪位置信息；

将所述弱监督训练样本数据输入至所述初始文本单字检测模型中，获得所述初始文本单字检测模型输出的预测位置信息和预测文字数量；

根据所述预测位置信息、所述预测文字数量、所述伪位置信息和标记文字数量确定所述弱监督损失值，以所述弱监督损失值为目标，得到所述迭代文本单字检测模型。

4.根据权利要求3所述的方法，所述基于所述初始文本单字检测模型和辅助单字检测算法生成所述弱监督训练样本数据对应的伪位置信息，包括：

将所述弱监督训练样本数据输入至所述初始文本单字检测模型中，获得所述初始文本单字检测模型输出的初始位置信息；

使用所述辅助单字检测算法得到所述弱监督训练样本数据对应的辅助位置数据；

根据所述初始位置信息和所述辅助位置信息确定所述伪位置信息。

5.根据权利要求4所述的方法，其特征在于，所述根据所述初始位置信息和所述辅助位置信息确定所述伪位置信息，包括：

针对所述弱监督训练样本数据中的每个弱监督样本图像，获取所述弱监督样本图像对应的初始位置信息的置信度；

若所述置信度高于设定的置信度阈值，则将所述弱监督样本图像对应的所述初始位置信息作为所述弱监督样本图像对应的伪位置信息；

若所述置信度不高于设定的置信度阈值，则将所述弱监督样本图像对应的所述辅助位置信息作为所述弱监督样本图像对应的伪位置信息。

6.根据权利要求3所述的方法，其特征在于，所述根据所述预测位置信息、所述预测文字数量、所述伪位置信息和标记文字数量确定所述弱监督损失值，包括：

根据所述预测位置信息和所述伪位置信息确定基础损失值，根据所述预测文字数量和所述标记文字数量确定损失值调节系数；

根据所述基础损失值和所述损失值调节系数确定所述弱监督损失值。

7.根据权利要求3所述的方法，其特征在于，所述初始文本单字检测模型包括锚点预测模块和位置预测模块，将所述弱监督训练样本数据输入至所述初始文本单字检测模型中，获得所述初始文本单字检测模型输出的预测位置信息，包括：

将所述弱监督训练样本数据输入至所述锚点预测模块中，获得所述锚点预测模块输出的至少一个初始锚点，其中，各所述初始锚点的高度相同；

将所述初始锚点输入至所述位置预测模块中，获得所述位置预测模块输出的所述预测位置信息。