摘要: 金属材料的质量问题直接关系到工业产品的安全性,传统的金属材料质量缺陷检测算法无法实现高精度的实时检测。为解决该问题,提出了一种针对金属表面缺陷检测框架 (YOLOv7-BCA)。首先采用动态稀疏采样来进行特征提取,增强网络对于细粒度的特征提取;接着设计特征增强模块,以实现3个相邻层的特征融合增强,使得定位信息更加准确;最后结合自适应空间特征融合思想,有效结合深度网络特征图的语义信息和浅层网络特征图的位置信息。实验结果表明,提出的YOLOv7-BCA平均检测精度为75.9%,比原始模型提升4.7%,证实了YOLOv7-BCA的性能在金属表面缺陷检测中具有较为明显的优势,可以对金属表面缺陷进行高度精确的定位,具有较强的实时性。
中图分类号: