基于改进的YOLOv5模型和射线法的车辆违停检测

作     者：庄建军 徐子恒 张若愚 ZHUANG Jianjun;XU Ziheng;ZHANG Ruoyu

作者机构：南京信息工程大学电子与信息工程学院南京210044 

出 版 物：《南京信息工程大学学报》 (Journal of Nanjing University of Information Science & Technology（Natural Science Edition）)

年 卷 期：2024年第16卷第3期

页      面：341-351页

学科分类：08[工学] 080203[工学-机械设计及理论] 0802[工学-机械工程] 

基　　金：国家自然科学基金(62171228) 江苏省研究生科研与实践创新计划(SJCX21_0354) 

主　　题：车辆违停检测 YOLOv5s算法 Ghost卷积 注意力机制 射线法 

摘      要：车辆违法停车将会降低道路通行效率,引发交通拥堵和交通事故.传统的车辆违停检测方法参数量大且准确度低.为此,本文提出了一种使用改进的YOLOv5模型和射线法的车辆违停检测方法.首先设计了轻量化的特征提取模块,减少模型参数量;其次在模型中加入注意力机制,从通道维度和空间维度增强模型的特征提取能力,保证模型精度;接着使用混合数据增强丰富数据集样本,提升复杂背景下的检测效果;然后选用EIoU作为损失函数提高模型定位能力.实验结果表明,改进后的模型均值平均精度达到91.35%,比原始YOLOv5s提升1.01个百分点,并且参数量减少35.79%.最后将改进后模型与射线法结合,在Jetson Xavier NX嵌入式平台的检测速度可以达到约28帧/s,能够实现实时检测.