- 专利标题: 一种基于MSA-Yolov5的恶劣天气下车辆检测方法及装置
-
申请号: CN202210794742.0申请日: 2022-07-07
-
公开(公告)号: CN115272987B公开(公告)日: 2023-08-22
- 发明人: 高尚兵 , 刘宇 , 李杰 , 陈浩霖 , 于永涛 , 张海艳 , 陈晓兵 , 张秦涛 , 胡序洋 , 李少凡
- 申请人: 淮阴工学院
- 申请人地址: 江苏省淮安市经济技术开发区枚乘东路1号
- 专利权人: 淮阴工学院
- 当前专利权人: 淮阴工学院
- 当前专利权人地址: 江苏省淮安市经济技术开发区枚乘东路1号
- 代理机构: 南京苏高专利商标事务所
- 代理商 柏尚春
- 主分类号: G06V20/54
- IPC分类号: G06V20/54 ; G06V10/82 ; G06V10/44 ; G06N3/0464 ; G06N3/08 ; G06N3/048
摘要:
本发明公开了一种基于MSA‑Yolov5的恶劣天气下车辆检测方法及装置,首先,将DAWN数据集划分为训练集与测试集,并进行预处理;其次,构建MSA‑Yolov5模型,包括Backbone模块、Neck模块以及Prediction模块;其中Backbone模块为MSA‑ResBlock,包括Focus网络、SCBAM模块和ResBlock模块;首先用SCBAM模块对恶劣天气下,模糊场景中的车辆、行人目标进行特征强化;然后通过多尺度特征融合,使用GIoU算法作为边界框损失函数,使用FocalLoss降低预测框内正负样本的不平衡问题;最后检测出恶劣天气下车辆、行人的位置信息和标签信息,并得出检测的精确度。本发明具有图像校正的计算量减小、检测流程精简、网络的识别精度高的特点,可以对恶劣天气下的车辆、行人进行检测,并且具有检测速度快、模型小、准确率高的优点。
公开/授权文献
- CN115272987A 一种基于MSA-Yolov5的恶劣天气下车辆检测方法及装置 公开/授权日:2022-11-01