一种基于特征增强的端到端缺陷检测方法

本发明涉及缺陷检测领域，具体是一种基于特征增强的端到端缺陷检测的方法，步骤1：采集缺陷图像；步骤2：将缺陷图像输入到Backbone网络进行特征提取；步骤3：将多尺度特征图归一化，然后经过全局查询细节增强模块得到输出；步骤4：将经过全局查询细节增强模块的输出得到的特征图S″i,j首先归一化，再经过特征融合模块得到对应的输出F″i,j,然后与S″i,j残差连接得到最后的输出Fi,j；步骤5：将Fi,5特征图输出至编码层进行尺度内交互；步骤6：将{Fi,3,Fi,4,F′i,5}输入值CCFM模块进行跨特征融合，得到FMi；步骤7：将特征向量FMi输入至IOU感知查询选择模块；步骤8：将FTopki特征输入至解码层输出预测的坐标框和类别；本发明解决跨尺度融合过程中语义缺失的问题，从而提高模型的检测能力。