基于改进YOLOv5的仿生机器鱼目标检测算法研究  

Research on Bionic Robotic Fish Object Detection Algorithm based on Improved YOLOv5

作　　者：王影[1] 孙可欣 刘振刚 高康盛 刘麒[1] WANG Ying;SUN Kexin;LIU Zhengang;GAO Kangsheng;LIU Qi(School of Information and Control Engineering,Jilin Institute of Chemical Technology,Jilin City 132022,China;Stauber(Hangzhou)Precision Mechanical Electronics Co.,Ltd.,Hangzhou 310018,China)

机构地区：[1]吉林化工学院信息与控制工程学院,吉林吉林132022 [2]史陶比尔(杭州)精密机械电子有限公司,浙江杭州310018

出　　处：《吉林化工学院学报》2024年第5期22-29,共8页Journal of Jilin Institute of Chemical Technology

基　　金：吉林市科技成果项目(201750244);吉林市科技成果项目(20190502118);吉林化工学院科研项目(2018064);吉林化工学院重大科技项目(2016033);吉林化工学院重大科技项目(2018017)。

摘　　要：为满足仿生机器鱼目标检测的需要,在YOLOv5基础上提出了一种轻量级检测算法,降低算法复杂度并提高精度。首先对YOLOv5s模型进行改进,通过GhostConv和C3Ghost模块降低参数量和计算量。其次,引入CA和CoordConv模块增强特征提取和目标位置感知能力,采用soft NMS减少使用传统非极大抑制(Non maximum suppression,NMS)带来的漏检、误检,同时使用MPDIoU简化相似性比较,提升检测精度和召回率。最后,所提出方法在目标检测数据集上的试验结果表明,改进的YOLOv5网络体积更小、精度更高,证明了该算法的有效性和优越性。To meet the needs of object detection for biomimetic robotic fish,a lightweight detection algorithm based on YOLOv5 was proposed to reduce algorithm complexity and improve accuracy.First,improvements were made to the YOLOv5s model by using GhostConv and C3Ghost modules to reduce the number of parameters and computational load.Second,CA and CoordConv modules were introduced to enhance feature extraction and target position perception capabilities,and soft NMS was used to reduce missed and false detections caused by traditional Non-Maximum Suppression(NMS).Additionally,MPDIoU was used to simplify similarity comparison,improving detection accuracy and recall rate.Finally,experimental results on the object detection dataset showed that the improved YOLOv5 network is smaller in size and higher in accuracy,demonstrating the effectiveness and superiority of the proposed algorithm.

关 键 词：改进YOLOv5 仿生机器鱼 目标检测 

分 类 号：TP391.41[自动化与计算机技术—计算机应用技术] TP242[自动化与计算机技术—计算机科学与技术]