基于改进YOLOv8多机械零件检测算法的研究

Research on a Multi-Mechanical Part Detection Algorithm Based on Improved YOLOv8

  • 摘要: 在模型架构方面,将原有的SPPF模块替换为具有更强特征提取能力的AIFI模块;其次在检测头部分引入动态头模块,增 强模型的特征表达能力和泛化性能,同时将原有的C2f模块优化为C2f-MLCA模块,进一步提升特征融合效果;最后在损 失函数设计上,采用新型的Wise-Inner-Shape IoU损失函数,有效提高了机械零件的定位精度。试验结果显示,改进后的 YOLOv8算法在机械零件检测任务中表现优异,查准率、召回率和平均精度分别为90.8%、84.2%、91.7%,平均精度较基 础模型提升了2.5个百分点,同时准确率和召回率均有所提升,充分验证了该算法在复杂机械零件检测中的实用性和优势。

     

    Abstract: improved YOLOv8-based mechanical part detection method. First,the original SPPF module in YOLOv8 is replaced with the AIFI module, which has stronger feature extraction capabilities. Then,a dynamic detection head (Dyhead) module is introduced in the detection head to enhance the modelʹs feature representation ability and generalization performance,while the original C2f module is optimized into the C2f-MLCA module,further improving feature fusion. Finally,a new Wise-Inner-Shape IoU loss function is used in the design of the loss function to effectively improve the localization accuracy of mechanical parts. Experimental results show that the improved YOLOv8 algorithm achieves significant results in mechanical part detection tasks,with Precision,Recall,and mAP@50 reaching 90.8%,84.2%, and 91.7%,respectively. The average precision improved by 2.5 percentage points compared to the base model,with both accuracy and recall also improved,fully verifying the superiority and practicality of the algorithm in complex mechanical part detection tasks.

     

/

返回文章
返回