[发明专利]文本检测方法、装置、电子设备及存储介质

权利要求书

1.一种文本检测方法，其特征在于，包括：

通过特征金字塔将原始光照图像转化为对应的参数矩阵，并基于所述参数矩阵调整所述原始光照图像的图像亮度，得到所述原始光照图像对应的待处理暗光图像；

基于特征提取网络对所述原始光照图像对应的原始暗光图像，以及所述待处理暗光图像分别进行多层特征提取，分别得到所述原始暗光图像对应的原始特征和补充特征；

基于所述补充特征对所述原始特征进行特征补全，得到特征补全的原始暗光图像；

基于输出网络输出所述特征补全的原始暗光图像的热力图和文本预测框，并基于所述热力图和所述文本预测框进行文本检测。

2.根据权利要求1所述的文本检测方法，其特征在于，所述基于所述热力图和所述文本预测框进行文本检测的步骤包括：

基于所述热力图确定所述特征补全的原始暗光图像中，各个像素点对应的像素分值；

基于各个所述像素分值确定对应的正样本区域，并基于所述正样本区域和所述文本预测框进行文本检测。

3.根据权利要求1所述的文本检测方法，其特征在于，所述原始特征包括第一部分原始特征和第二部分原始特征，

所述基于所述补充特征对所述原始特征进行特征补全，得到特征补全的原始暗光图像的步骤包括：

将所述补充特征和所述第一部分原始特征，输入至第一特征融合模块，其中，所述第一特征融合模块包括第一合并函数和第一卷积层；

基于所述第一合并函数和所述第一卷积层，将所述补充特征补全至所述第一部分原始特征中；

将所述第二部分原始特征及其对应的上一级特征进行特征融合，得到所述特征补全的原始暗光图像。

4.根据权利要求3所述的文本检测方法，其特征在于，所述第二部分原始特征包括第一特征和第二特征，所述第一特征在所述第二特征的底层，

所述将所述第二部分原始特征及其对应的上一级特征进行特征融合，得到所述特征补全的原始暗光图像的步骤包括：

获取所述第一特征对应的第一上一级特征，将所述第一特征和所述第一上一级特征输入至第二特征融合模块，其中，所述第二特征融合模块包括第二合并函数和第二卷积层；

基于所述第二合并函数和所述第二卷积层，将所述第一特征和所述第一上一级特征进行第一次特征融合；

获取所述第二特征对应的第二上一级特征，基于所述第二合并函数和所述第二卷积层，将所述第二特征和所述第二上一级特征进行第二次特征融合；

基于所述第一次特征融合和所述第二次特征融合，得到所述特征补全的原始暗光图像。

5.根据权利要求1所述的文本检测方法，其特征在于，所述基于所述参数矩阵调整所述原始光照图像的图像亮度，得到所述原始光照图像对应的待处理暗光图像的步骤包括：

确定所述原始光照图像中各个图像点的像素值，并基于所述参数矩阵结合预设迭代函数对各个所述像素值进行迭代调整，得到各个所述像素值对应的调整后的像素值；

基于各个所述调整后的像素值，及平均亮度的约束信息调整所述原始光照图像的图像亮度，得到所述待处理暗光图像。

6.根据权利要求1所述的文本检测方法，其特征在于，所述基于特征提取网络对所述原始光照图像对应的原始暗光图像进行多层特征提取，得到所述原始暗光图像对应的原始特征的步骤包括：

基于所述特征提取网络对原始暗光图像进行多层特征提取，得到原始暗光图像对应的预设层特征，并将所述预设层特征确定为所述原始特征。

7.根据权利要求1至6任一项所述的文本检测方法，其特征在于，所述通过特征金字塔将原始光照图像转化为对应的参数矩阵，并基于所述参数矩阵调整所述原始光照图像的图像亮度，得到所述原始光照图像对应的待处理暗光图像的步骤之前，还包括：

通过摄像设备拍摄在预设正常光照环境下带有文本数据的图像，得到对应的原始光照图像；

通过所述摄像设备拍摄在预设暗光环境下所述原始光照图像对应的图像，得到所述原始光照图像对应的原始暗光图像。