基于多头注意力机制的改进组卷积和特征增强融合网络的轧钢表面缺陷检测方法

本发明提供基于多头注意力机制的改进组卷积和特征增强融合网络的轧钢表面缺陷检测方法，其中，方法包括：获取轧钢表面缺陷图像数据集；基于分组空间多尺度卷积模块和特征增强融合网络，根据轧钢表面缺陷图像数据集，训练缺陷检测模型；将所需检测图像输入缺陷检测模型，获得目标轧钢表面缺陷检测结果。本发明的基于多头注意力机制的改进组卷积和特征增强融合网络的轧钢表面缺陷检测方法，引入分组空间多尺度卷积模块和特征增强融合网络改进YOLOv7目标检测模型，获取预训练缺陷检测模型，基于训练样本训练预训练缺陷检测模型获取缺陷检测模型，将所需检测图像输入缺陷检测模型，获得缺陷检测结果，提升了模型缺陷检测性能。