公开(公告)号：CN118469984A

公开(公告)日：2024-08-09

申请号：CN202410689921.7

申请日：2024-05-30

Applicant: 常州大学

Inventor： 吴鹏 ,  李旻 ,  袁伟 ,  许浩 ,  陈志豪

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

Abstract: 本申请涉及缺陷检测技术领域，尤其涉及一种基于YOLO v8改进模型的金属表面缺陷的检测方法，包括以下步骤：建立金属表面数据集，数据集包括不同的表面缺陷种类；建立YOLO v8n网络模型，调整和配置相应的参数；建立YOLO v8n改进模型，利用RCS模块替换C2f模块，在模型中引入注意力机制，用Rhead替换Detect模块，优化卷积神经网络中neck部分的结构；训练和优化YOLO v8n改进模型，使用训练优化后的YOLO v8n改进模型对金属表面缺陷进行检测。本申请在YOLO v8n模型中利用RCS模块替换C2f模块，在模型中引入注意力机制，用Rhead替换Detect模块优化卷积神经网络中neck部分的结构，从而提高了缺陷检测的检测精度与速度。