一种基于改进YOLOv7的钢材表面缺陷检测方法

    公开(公告)号:CN117237611A

    公开(公告)日:2023-12-15

    申请号:CN202311315577.7

    申请日:2023-10-11

    Abstract: 本发明涉及数字图像处理目标识别技术领域,具体说是一种基于改进YOLOv7的钢材表面缺陷检测方法,包括:在主干网络中,通过引入坐标卷积CoordConv,裁剪CBS层,采用SPPF结构和Biformer注意力机制重构SPPCSPC,减少计算量和参数量,提取了更多小目标特征,提高了本方法对钢材表面缺陷检测的精度;通过添加小目标检测层,将高分辨率特征图与原来三个尺度的特征图进行特征融合,提升小目标特征捕获量,并增加对应的检测头,提高了小目标检出率和准确率,提高了本方法对钢材表面缺陷检测的精度;采用新的上采样操作CARAFE,增大感受野,减少小目标特征的丢失,减小了钢材表面缺陷的漏检率;通过将损失函数由边框拟合CIoU替换为动态非单调聚焦WiseIoU,提升网络模型收敛速度。

Patent Agency Ranking