融合改进的YOLOv5n和通道剪枝的寄生卵检测和分类  

Detection and Classification of Parasitic Eggs Using a Fused Improved YOLOv5n and Channel Pruning Algorithm

作　　者：王杰 马纪颖 WANG Jie;MA Ji-ying(School of Computer Science and Technology,Shenyang University of Chemical Technology,Shenyang 110142,China;Liaoning Key Laboratory of Intelligent Technology for Chemical Process Industry,Shenyang 110142,China)

机构地区：[1]沈阳化工大学计算机科学与技术学院,辽宁沈阳110142 [2]辽宁省化工过程工业智能化技术重点实验室,辽宁沈阳110142

出　　处：《计算机技术与发展》2025年第2期146-152,共7页Computer Technology and Development

基　　金：辽宁省自然基金项目(2022-MS-291);国家外国专家项目计划(G2022006008L);辽宁省教育厅基本科研项目(LJKMZ20220781)。

摘　　要：当前众多目标检测模型过于复杂,难以实现将寄生卵的检测和分类任务部署在移动设备,就此该文研究探讨了一种融合改进的YOLOv5n和通道剪枝的算法。选择YOLOv5是由于YOLOv5的轻量化以及较高的精确度,能够达到该文的实验目的。该文采用融合C3_Faster模块和RepConv重参数化模块对YOLOv5n的BackBone中的所有C3模块和Neck网络中部分卷积模块进行替换,C3_Faster模块通过PConv减少卷积操作加快网络模型推理速度,RepConv重参数化模块在训练阶段实行多分支结构增强特征提取能力,在验证阶段实行单分支结构加快检测速度,同时在改进后的YOLOv5n模型上进行稀疏训练和通道剪枝,通过减少模型中的冗余通道来降低模型复杂度、减少参数数量、提高检测效率和降低模型权重。在寄生卵检测和分类任务对比实验中,该方法与YOLOv5n、YOLOv5s、YOLOv7-tiny、YOLOv8n和SSD目标检测算法相比,在检测精度略微下降的情况下,在GFLOPs、FPS、参数数量以及模型权重上具有相对优势。经过实验验证,模型检测精度保持98.3%的同时能够更方便更容易部署在性能不高的移动设备。该文为基于YOLOv5n的寄生卵检测和分类任务在实用性方面提供了一种有效的解决方案。Many current target detection models are too complex to implement parasitic egg detection and classification tasks on mobile devices.Therefore,we study an algorithm that integrates improved YOLOv5n and channel pruning.YOLOv5 is selected because of its lightweight and high accuracy,which can achieve the experimental purpose in this paper.The fusion C3_Faster module and RepConv heavy parameterization module are used to replace all C3 modules in BackBone of YOLOv5n and some convolutional modules in Neck network.C3_Faster module reduces convolutional operations by PConv to speed up network model inference.The RepConv heavy parameterization module implements multi-branch structure to enhance feature extraction capability in the training stage,and single-branch structure to accelerate detection speed in the verification stage.Meanwhile,sparse training and channel pruning are carried out on the improved YOLOv5n model.By reducing the redundant channels in the model,the complexity of the model is reduced,the number of parameters is reduced,the detection efficiency is improved and the model weight is reduced.Compared with YOLOv5n,YOLOv5s,YOLOv7-tiny,YOLOv8n and SSD target detection algorithms,the proposed method has comparative advantages in GFLOPs,FPS,number of parameters and model weight under the condition of slightly reduced detection accuracy.After experimental verification,the model detection accuracy remains 98.3%,and it can be more convenient and easy to deploy on mobile devices with low performance.We provide an effective solution for the parasitic egg detection and classification task based on YOLOv5n in terms of practicality.

关 键 词：YOLOv5算法 寄生卵检测 C3_Faster RepConv 通道剪枝 

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