重庆理工大学学报(自然科学) ›› 2024, Vol. 38 ›› Issue (8): 167-174.

• 信息·计算机 • 上一篇    下一篇

改进 YOLOv5s的小样本 3D打印点阵结构表面缺陷检测

安治国,鲜青霖,许 亮   

  1. 重庆交通大学 机电与车辆工程学院,重庆 40007
  • 出版日期:2024-09-09 发布日期:2024-09-09
  • 作者简介:安治国,男,博士,副教授,主要从事人工智能研究,Email:anzhiguo@cqjtu.edu.cn。

  • Online:2024-09-09 Published:2024-09-09

摘要: 3D打印点阵结构已经广泛应用于航空航天、机械和建筑等行业,其表面缺陷分布不均匀且特征微弱,常常造成漏检和误检。针对这一问题,提出了一种 YOLOv5sPD模型。在该模型中,添加了 XSPPF模块和空洞金字塔池化模块,提高了模型对不同缺陷特征的获取能力;针对 3D打印点阵结构表面缺陷分布杂乱导致误检率高的问题,在YOLOv5s模型中加入了 ECA模块;考虑到 3D打印点阵结构表面缺陷尺寸信息无规律并且差异较大而导致的预测框与真实框间方向不一致,采用了 SIoU损失函数。采用改进模型对制作的 3D打印点阵结构表面缺陷数据集进行检测,结果表明:缺陷检测的召回率达到 94.0%,平均精度 mAP@0.5达到 96.2%,所提出的改进算法可以实现对 3D打印点阵结构表面缺陷自动检测。

关键词: 3D打印, 点阵结构, YOLOv5s, 缺陷检测, 平均精度

中图分类号: 

  • TP391.4