文献类型：期刊文章

作　　者：董恒祥[1] 潘江如[2] 董芙楠[2] 赵晴[1] 郭鸿鑫[1]

DONG Hengxiang;PAN Jiangru;DONG Funan;ZHAO Qing;GUO Hongxin(School of Traffic and Logistics Engineering,Xinjiang Agricultural University,Urumqi 830000,China;College of Control Engineering,Xinjiang Institute of Engineering,Urumqi 830000,China)

机构地区：[1]新疆农业大学交通与物流工程学院,新疆乌鲁木齐830000 [2]新疆工程学院控制工程学院,新疆乌鲁木齐830000

出　　处：《现代电子技术》

年　　份：2024

卷　　号：47

期　　号：3

起止页码：179-186

语　　种：中文

收录情况：BDHX、BDHX2023、JST、RCCSE、ZGKJHX、核心刊

摘　　要：针对车辆检测中边界框回归损失函数与检测目标尺度不匹配导致的误检、漏检以及精度较低等问题,基于YOLOv5s模型对4种有代表性的边界框回归损失函数进行对比实验,并在UA-DETRA、VisDrone2019、KITTI数据集上进行验证,利用漏检率、误检率、准确率、召回率、mAP@0.5、迭代过程的边界框损失值以及目标检测结果对其适用场景进行分析研究。结果显示:CIoU整体性能最差;SIoU在KITTI数据集上整体性能最优,准确率最高,达到了94.5%,漏检率降到了1.2%,适用于中尺度目标检测任务;Focal-EIoU在VisDrone2019数据集中各项指标远优于其他损失函数,召回率和mAP@0.5指标相较于CIoU分别提高了1.6%和1.8%,误检率降低了6.9%,且迭代过程损失值远低于其他损失函数,适用于小尺度目标检测任务;WIoU在UA-DETRA数据集整体性能最优,漏检率、召回率以及mAP@0.5指标优于其他损失函数,适用于大尺度目标检测任务。此研究为目标检测任务的边界框回归损失函数的选择提供了重要的基础。

关 键 词：车辆检测 边界框回归损失函数  目标尺度  YOLOv5s  CIoU  SIoU  Focal-EIoU  WIoU  

分 类 号：TN911.1-34] TP391]