基于改进Shuffle-RetinaNet的红外车辆检测算法 |
| |
引用本文: | 范晓畅,梁煜,张为.基于改进Shuffle-RetinaNet的红外车辆检测算法[J].激光与光电子学进展,2023(24):118-127. |
| |
作者姓名: | 范晓畅 梁煜 张为 |
| |
作者单位: | 天津大学微电子学院 |
| |
摘 要: | 针对当前红外场景下多尺度车辆检测精度欠佳且算法模型复杂度高的问题,提出了基于Shuffle-RetinaNet的红外车辆检测算法。该算法以RetinaNet网络为基础,并选用ShuffleNetV2作为特征提取网络。提出双分支注意力模块,通过双分支结构和自适应融合方法增强网络对红外图像中目标关键特征的提取能力;优化特征融合网络,集成双向交叉尺度连接和快速归一化融合,增强目标多尺度特征的表达能力;设置校准因子增强分类和回归之间的任务交互,提高目标分类和定位的准确性。该算法在自建红外车辆数据集上的检测精度达到92.9%,参数量为11.74×106,浮点计算量为24.35×109,同时在公开红外数据集FLIR ADAS上也展现出较好的检测性能。实验结果表明:该算法具有较高的检测精度,且模型复杂度低,在红外车辆检测领域具有较高的应用价值。
|
关 键 词: | 目标检测 红外车辆 通道注意力模块 多尺度特征融合 校准因子 |
|
|