针对于雾天交通场景下道路目标检测的准确性亟待提升的问题,提出一种基于改进YOLOv7-tiny算法。首先,将YOLOv7-tiny中引入上采样算子CARAFE(Content-Aware ReAssembly of Features)模块,以此来提升模型的特征融合能力;将ELAN-tiny模块...针对于雾天交通场景下道路目标检测的准确性亟待提升的问题,提出一种基于改进YOLOv7-tiny算法。首先,将YOLOv7-tiny中引入上采样算子CARAFE(Content-Aware ReAssembly of Features)模块,以此来提升模型的特征融合能力;将ELAN-tiny模块与多元分支模块DBB(Diverse Branch Block)融合,以增强模型提取特征信息的能力,提升目标检测精度。改进算法在保证轻量化的前提下,能够有效地提升对雾天道路目标的检测性能。实验结果表明,在模型参数量和计算量少量增加的情况下,改进算法相较于基线算法,mAP@50和mAP@50∶95分别提升了1.0%和0.5%,论证了改进算法的有效性。展开更多
基金supported by Key Research and Development Project of China(No.2021YFB1600104)National Natural Science Foundation of China(No.52002031)Scientific Research Project of Shaanxi Provincial Department of Transportation(No.20-24K,20-25X)。
文摘为了提高多尺度道路目标的检测性能,本文针对目标检测算法在非极大值抑制阶段的检测质量表征不合理问题,提出了一种优化定位置信度改进算法.首先基于RepPoints构建研究框架,研究定位置信度对多尺度道路目标的敏感性.在敏感性研究结果的基础上,本文提出了混合定位置信度.然后针对IoU定位置信度无法区分重叠程度相同的包围框的缺陷,提出了CIoU定位置信度.最后将这两种定位置信度结合得到优化定位置信度改进算法,解决了检测质量表征不合理问题.在道路场景数据集Cityscapes上的实验结果表明,混合定位置信度和CIoU定位置信度单一作用时均有效,共同作用时精度提高2.4%,多尺度目标检测精度均有显著提升,且实时性没有下降.相较于主流道路场景检测算法如Cascade-RCNN、FCOS等,本文算法取得了最高的mAP、AP M和AP L.