基于注意力特征金字塔机制的小样本钢材缺陷检测方法

本发明公开一种基于注意力特征金字塔机制的小样本钢材缺陷检测方法，先经由卷积神经网络获得初始多尺度特征图，再采用自上向下和横向连接方式，利用改进的多尺度注意力网络对初始多尺度特征图进行逐一处理，获得中间多尺度特征图，然后采用自下向上和横向连接方式，利用改进的多尺度注意力网络对中间多尺度特征图进行逐一处理，获得最终多尺度特征图，最后输出最终分类结果；采用伪标签半监督学习算法对检测网络模型进行训练，利用局部线性嵌入算法对训练输出的高层特征进行降维与可视化处理，筛选出误标错标的图像去掉标签放入未标记的图像数据，重新训练生成伪标签，以扩充训练集，直至达到训练要求；将待检图像输入训练好的检测网络模型识别。