摘要

近年来，基于分割的文本检测方法在场景文本检测领域受到广泛关注，因为分割结果能够更准确地描述各种形状的文本，例如弯曲文本。然而，基于分割的检测方法通常依赖于二值化后处理步骤，该步骤将分割网络生成的概率图转换为文本的边界框或区域。本文提出一种名为可微分二值化（Differentiable Binarization, DB）的模块，可将二值化过程直接嵌入到分割网络中。通过与DB模块联合优化，分割网络能够自适应地设定二值化阈值，不仅简化了后处理流程，还显著提升了文本检测的性能。基于一个简单的分割网络架构，我们在五个基准数据集上验证了DB模块的有效性，结果表明其在检测精度与速度方面均持续达到当前最优水平。尤其值得注意的是，在采用轻量级主干网络的情况下，DB带来的性能提升尤为显著，从而有助于在检测精度与计算效率之间实现理想的权衡。具体而言，在MSRA-TD500数据集上，使用ResNet-18作为主干网络的检测器取得了82.8的F-measure，推理速度达到62 FPS。代码已开源，地址为：https://github.com/MhLiao/DB