首页|基于改进YOLOv5s汽车驾舱遗忘物检测

基于改进YOLOv5s汽车驾舱遗忘物检测

扫码查看
针对目前汽车驾舱内遗忘物检测精度不高、速度慢和检测效果差的问题,提出一种基于改进的YOLOv5s汽车驾舱遗忘物的检测方法;该检测方法将YOLOv5s作为基础网络,在此基础上进行改进;首先本实验将SE注意力模块添加到Backbone网络中,加强模型对通道信息的关注提升目标检测性能;其次改进空间金字塔池化模块,将原有的SPPF模块改进为SPPCSPC模块,使得该网络更加关注待检测目标的特征;最后同时引入GSConv层,能够缓解DSC(深度可分离卷积)的缺陷,并充分利用DSC的优势,在小目标检测方面取得明显的提升效果,既保证了语义信息又平衡了模型的准确性,也提升了检测速度;通过训练结果说明,改进后的网络与原YOLOv5s网络相比,其平均精度均值mAP提高了 2%,查准率提升了 3。5%;改进后的网络具有良好的提升效果,表明了该方法的有效性。
Automotive Cockpit Oblivion Detection Based on Improved YOLOv5s
Aimed at the low detection accuracy,slow speed and poor detection of amnesia in current automobile cockpits,an im-proved YOLOv5s automotive cockpit oblivion detection method forgetting is proposed.The detection method uses YOLOv5s as a basic network and improves on this basis.Firstly,the SE attention module is added to a backbone network to strengthen the model's atten-tion to the channel information and improve the target detection performance.Secondly,the space pyramid pool module is improved,and the original SPPF module is improved to the SPPCSPC module,which makes the network pay more attention to the characteristics of the target to be detected.Finally,the GSConv layer is simultaneously introduced to alleviate the detects of depth wise separable convolution(DSC)and fully utilize the advantages of the DSC to achieve significant results in detecting small targets.It ensures the semantic information,balances the accuracy of the model,and improves the detection speed.Experimental results show that com-pared with the original YOLOv5s network,the average mean average precision(mAP)of the improved network is increased by 2%,and the detection accuracy by 3.5%.The improved network has a good improvement effect,which shows the effectiveness of the method.

oblivionYOLOv5sattention mechanismSEGSConvSPPCSPC

吴继薇、焦良葆、焦波、祝阳、高阳

展开 >

南京工程学院人工智能产业技术研究院,南京 211167

江苏省智能感知技术与装备工程研究中心,南京 211167

遗忘物 YOLOv5s 注意力机制 SE GSConv SPPCSPC

江苏省自然科学基金资助项目江苏省政策引导类计划项目

BK20201042SZ-SQ2020007

2024

计算机测量与控制
中国计算机自动测量与控制技术协会

计算机测量与控制

CSTPCD
影响因子:0.546
ISSN:1671-4598
年,卷(期):2024.32(9)
  • 11