基于双条带窗口金字塔网络模型的热轧钢表面缺陷检测方法和装置

本发明公开一种基于双条带窗口金字塔网络模型的热轧钢表面缺陷检测方法和装置，包括：步骤1、收集热轧钢表面缺陷图像，构建缺陷图像集，并对缺陷图像集中的图像进行数据增强处理，得到增强缺陷数据集；步骤2、采用CSPDarkNet53作为骨干网络，在其后端加入DSwinTransformer模块，同时将原有的Neck部分替换为BiFPN，并在检测头前加入多头上下文集成注意力机制模块，以构建双条带窗口金字塔网络；步骤3、将增强缺陷数据集输入训练好的DSWPN网络，基于CIOU损失函数进行训练，直至损失函数收敛，完成缺陷检测模型训练；步骤4、实时采集热轧钢表面图像，输入训练好的DSWPN网络，得到热轧钢表面缺陷检测结果。