《计算机工程与应用杂志》发表论文赏析
作者:雷景生, 章志豪, 钱小鸿, 王巍然, 杨胜英
单位:1.浙江科技学院 信息与电子工程学院,杭州 310023;2.国网上海市电力公司信息通信公司,上海 200072
摘要:针对现有的车牌检测算法在复杂环境下检测多方向车牌效果不佳、实时性差以及模型参数量和计算量过大等问题,提出了一种基于YOLOX的轻量级多方向车牌检测算法。通过调整残差组件的数量,并采用大卷积核结合深度可分离卷积的方法,降低了主干网络的参数量。引入通道注意力机制,以有效提取通道交互信息,减少噪声的干扰。使用深度可分离卷积和调整扩展率的方法对特征融合网络进行轻量化处理。设计了旋转解耦头,通过添加角度预测分支,使其能够更精准地预测多方向车牌的旋转边界框。采用旋转IoU损失代替水平IoU损失,提高检测的准确性。CCPD数据集上的实验结果表明,改进算法的参数量和计算复杂度分别为2.38×106和12.97?GFLOPs,相较于YOLOX-tiny分别减少了45%和33%,检测精度AP70为94.9%,每秒检测帧数为76.6?FPS。改进后的车牌检测模型能够在保持高精度的同时实时检测多方向车牌。
关键词:深度学习,轻量化,YOLOX,目标检测,注意力机制