摘要： 工业生产过程中，钢材表面缺陷的检测对于钢材的质量控制发挥着十分重要的作用，针对钢材表面缺陷检测中存在的检测精度低、检测速度慢等问题，提出了一种钢材表面缺陷检测的改进SSD算法。在所提算法中，采用Transformer多头注意力机制模块代替原SSD结构中的Conv5＿1层，以提高小目标检测的能力，原SSD结构中的Conv7操作替换为Involution算子操作，以减少运算的参数量，对网络结构进行特征融合处理，以更全面地检测特征图中所包含的信息。利用NEU-DET数据集进行实验，实验结果表明改进后的SSD算法是有效的，可以高效检测到钢材表面的小目标缺陷，相比改进前平均检测精度提高了4.5%,检测速度提高了13.6%。

关键词: 钢材表面缺陷检测， 改进SSD算法， 注意力机制， Involution算子， 特征融合

中图分类号: 

• TP391.41，TG142.1