公开(公告)号：CN118134850A

公开(公告)日：2024-06-04

申请号：CN202410151859.6

申请日：2024-02-02

Applicant: 安徽工业大学

Inventor： 黄俊 ,  何乐 ,  郑啸 ,  洪旭东

IPC: G06T7/00 ,  G06V10/774 ,  G06V10/776 ,  G06V10/82 ,  G06N3/0464 ,  G06N3/08 ,  G06V10/764

Abstract: 本发明公开了基于YOLOv5改进的钢铁表面缺陷检测方法及装置，涉及图像缺陷检测技术领域，解决了钢铁检测中微小目标技术检测精度不高的问题；本发明通过获取大量的钢铁表面缺陷图片数据对源图片数据进行预处理与类别标注，构建钢铁表面缺陷数据集；以及基于YOLOv5算法改进的钢铁表面缺陷检测模型，有效地引入BiFormer注意力机制并改进CSP模块，提高对微小目标的检测能力，使得模型提高了对钢铁表面缺陷的检测精度，有助于工作人员方便、快捷地判断钢铁瑕疵情况，同时也解决了端到端检测算法检测精度低于分步检测算法的问题。