基于Swin-Transformer网络结构的钢板表面缺陷检测方法及电子设备

本发明公开基于Swin‑Transformer网络结构的钢板表面缺陷检测方法及电子设备：用每层Stage对图像尺度特征提取并输出特征图；用不同卷积核对特征图卷积得到尺度特征图；对尺度特征图全局池化提取池化向量，根据池化向量生成融合尺度权重，通过加权融合得融合特征图；对融合特征图操作，引入融合尺度权重得分类分支特征并加权融合，输入分类全连接层得到分类预测结果；对融合特征图全局平均池化，得全局特征表示，对池化向量处理，将融合尺度权重融入形成回归特征表示并输入回归全连接层得回归预测结果；将分类特征表示与全局特征表示特征拼接，生成交互权重，对分类预测结果和回归预测结果通过加权得最终分类输出和最终回归输出。本发明解决表面缺陷检测问题。