arXiv:2411.02794cs.CV2024-11被引 11

提出相似掩码与特征修正模块,实现交通、工业和自然场景下实时文本检测

Real-Time Text Detection with Similar Mask in Traffic, Industrial, and Natural Scenes

  • 用相似掩码保留文本几何信息,减少后处理时间50%
  • 通过特征修正模块提升准确率,在多个数据集上达到领先性能
  • 构建含运动模糊的交通文本数据集,增强模型泛化能力

智能交通场景中的文本包含大量信息,充分挖掘这些信息是推动智能交通发展的重要动力。与通用场景不同,交通文本检测还需兼顾快速推理速度与高精度。现有实时方法多基于收缩掩码,损失部分几何语义信息且依赖复杂后处理。此外,以往方法仅关注最终输出正确性,忽略中间过程的特征优化。为此,本文提出高效多场景文本检测器SM-Net,包含有效文本表示相似掩码(SM)与特征修正模块(FCM)。SM旨在最大程度保留实例几何信息,其后处理节省50%时间,精准重建文本轮廓;FCM促使误检特征远离正样本特征中心,从特征层面优化预测。消融实验验证了SM效率与FCM有效性。针对现有交通数据集标注质量低、闭源等问题,我们构建并标注了含运动模糊的交通文本数据集。在交通、工业和自然场景数据集上的实验表明,SM-Net在多个基准上达到当前最优(SOTA)性能。代码与数据集已开源。

原文摘要 · Abstract (English)

Texts on the intelligent transportation scene include mass information. Fully harnessing this information is one of the critical drivers for advancing intelligent transportation. Unlike the general scene, detecting text in transportation has extra demand, such as a fast inference speed, except for high accuracy. Most existing real-time text detection methods are based on the shrink mask, which loses some geometry semantic information and needs complex post-processing. In addition, the previous method usually focuses on correct output, which ignores feature correction and lacks guidance during the intermediate process. To this end, we propose an efficient multi-scene text detector that contains an effective text representation similar mask (SM) and a feature correction module (FCM). Unlike previous methods, the former aims to preserve the geometric information of the instances as much as possible. Its post-progressing saves 50$\%$ of the time, accurately and efficiently reconstructing text contours. The latter encourages false positive features to move away from the positive feature center, optimizing the predictions from the feature level. Some ablation studies demonstrate the efficiency of the SM and the effectiveness of the FCM. Moreover, the deficiency of existing traffic datasets (such as the low-quality annotation or closed source data unavailability) motivated us to collect and annotate a traffic text dataset, which introduces motion blur. In addition, to validate the scene robustness of the SM-Net, we conduct experiments on traffic, industrial, and natural scene datasets. Extensive experiments verify it achieves (SOTA) performance on several benchmarks. The code and dataset are available at: \url{https://github.com/fengmulin/SMNet}.

文本检测实时系统交通视觉特征优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。