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