基于改进YOLOv5的退役轴类零件表面损伤检测方法

刘伟嵬, 邱佳鹤, 胡光大, 刘泽远

PDF(3893 KB)
PDF(3893 KB)
东北大学学报(自然科学版) ›› 2024, Vol. 45 ›› Issue (07) : 1002-1010.

基于改进YOLOv5的退役轴类零件表面损伤检测方法

  • 刘伟嵬, 邱佳鹤, 胡光大, 刘泽远
作者信息 +
History +

摘要

针对传统检测方法在对退役轴类零件进行损伤检测时存在效率低、结果一致性差等问题,提出了一种基于改进YOLOv5的退役轴类零件表面损伤检测方法 .首先,将注意力机制嵌入检测算法中,增强了损伤在图像中的特征表示;然后,采用重复加权双向特征融合方法改进了检测模型的网络结构,有效提升了网络特征提取能力;最后,使用Ghostconv卷积模块代替普通卷积,大幅度降低了模型参数量.实验结果显示,改进后的算法模型精度比原始YOLOv5提升了6.9%,达到88.4%,同时模型参数量减少了6.1%,保证了检测速度与YOLOv5持平.与YOLOv3,SSD,Faster-RCNN等主流检测方法相比,在保证较高检测速度的同时,检测精度也有着明显优势.

关键词

YOLOv5 / 表面损伤检测 / 注意力机制 / 多路特征融合 / Ghostconv

中图分类号

TP391.41 / TH133.2

引用本文

导出引用
刘伟嵬, 邱佳鹤, 胡光大, 刘泽远. 基于改进YOLOv5的退役轴类零件表面损伤检测方法. 东北大学学报(自然科学版). 2024, 45(07): 1002-1010

基金

国家自然科学基金资助项目(52175455,51975100)

评论

PDF(3893 KB)

Accesses

Citation

Detail

段落导航
相关文章

/