公开(公告)号：CN118799553A

公开(公告)日：2024-10-18

申请号：CN202410891153.3

申请日：2024-07-04

Applicant: 国核电力规划设计研究院有限公司

Inventor： 刘超 ,  程宇航 ,  牛章玉 ,  高桂棠 ,  杨舒越 ,  董斌 ,  李鹤峰 ,  张明 ,  赵瑞 ,  王方

IPC: G06V10/25 ,  G06V20/17 ,  G06V10/82 ,  G06N3/045 ,  G06N3/0464 ,  G06N3/082 ,  G06V10/80

Abstract: 本公开提出一种金具缺陷识别方法及装置，涉及目标检测技术领域。其中，方法包括：获取金具的图像信息；将图像信息输入至金具检测模型，获得金具的缺陷检测结果；金具检测模型基于YOLO v7网络结构构建，金具检测模型包括主干网络、路径聚合网络PANet和检测头Head，其中，主干网络为Shufflenetv2网络，PANet与检测头Head之间增加卷积块注意力模块CBAM。本公开利用Shufflenetv2网络作为YOLO v7网络结构为基础的主干网络，使模型更加轻量化，加快检测速度。通过引入卷积块注意力模块CBAM加强模型对小目标的检测能力，提高了在关键金具类的小尺度目标检测方面的质量与效率。本公开可在提升检测速率的同时在一定程度上保证目标检测的精度。