基于改进DBNet和SVTR算法的连铸板坯号检测与识别
Detection and recognition of continuous casting slab numbers based on improved DBNet and SVTR algorithms
刘乐 1张晓松 1黄锋 1方一鸣1
作者信息
- 1. 燕山大学智能康复及神经调控河北省重点实验室 秦皇岛 066004
- 折叠
摘要
针对钢铁连铸产线板坯号识别字符区域小、光照变化复杂、板坯号图像质量差等问题,提出了一种基于深度学习的连铸板坯号检测与识别两阶段算法.首先,基于采集的连铸产线板坯图像,制备用于板坯号检测与识别的数据集;其次,在板坯号检测阶段,基于DBNet算法设计一种AD-PAN特征融合结构,以增强检测算法的多尺度特征融合能力和扩大感受野,提高板坯号定位精度;再次,在板坯号识别阶段,引入SPIN矫正网络和SVTR板坯号识别网络进行端到端训练,使其能够主动转换输入亮度,并改善字符间以及字符与背景间色彩失真的问题.最后,在自制的板坯号检测与识别数据集上进行了对比实验.实验结果表明,本研究提出的算法能够有效定位辊道上不同位置的板坯,并且在复杂背景下对板坯号进行鲁棒识别.其中,板坯号检测Hmean数值为 97.92%,板坯号识别的准确率为 97.33%,验证了本文所提算法具有较高的板坯号检测与识别精度.
Abstract
In order to tackle the challenges associated with small character regions,complex lighting variations,and poor image quality in the identification of slab numbers in steel continuous casting production lines,a two-stage algorithm is proposed for slab number detection and recognition,utilizing deep learning techniques.Firstly,datasets for slab number detection and recognition are prepared based on the collected slab images of a continuous casting production line.Secondly,in the slab number detection stage,an AD-PAN feature fusion structure based on the DBNet algorithm is designed to enhance the multi-scale feature fusion capability and expand the receptive field of the detection algorithm,thereby improving the localization accuracies of the slab numbers.Thirdly,in the slab number recognition stage,the proposed algorithm incorporates a SPIN correction network and a SVTR slab number recognition network for end-to-end training,enabling them to actively transform input brightness and improve color distortion among characters and between characters and backgrounds.Finally,comparative experiments are conducted on self-made datasets for slab number detection and recognition,and experimental results demonstrate that the algorithm proposed in this study is capable of effectively locating slab at different positions on the roller table and robustly recognizing slab numbers in complex backgrounds.The Hmean value for slab number detection is 97.92%,and the accuracy for slab number recognition is 97.33%,confirming the high accuracy of slab number detection and recognition achieved by the proposed algorithm.
关键词
板坯号识别/DBNet/特征金字塔融合/端到端网络/SPIN矫正/SVTRKey words
slab number recognition/DBNet/feature pyramid fusion/end-to-end network/SPIN correction/SVTR引用本文复制引用
基金项目
秦皇岛市科学技术研究与发展计划(202302B048)
河北省创新能力提升计划(22567619H)
河北省教育厅科研项目(ZD2021102)
出版年
2024
NETL
NSTL
万方数据