公开(公告)号：CN117893517A

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

申请号：CN202410096457.0

申请日：2024-01-24

Applicant: 南京信息工程大学

Inventor： 张凯旋 ,  杜景林

IPC: G06T7/00 ,  G06V10/82 ,  G06V10/32 ,  G06V10/25 ,  G06V10/44 ,  G06V10/774 ,  G06V10/776 ,  G06N3/0464 ,  G06N3/045 ,  G06N3/048

Abstract: 本申请公开了基于改进YOLOv5的布匹缺陷检测方法，属于图像处理和模式识别技术领域，包括以下步骤；获取待检测布匹图像；将所述待检测布匹图像输入至训练好的改进YOLOv5模型中，获得缺陷检测结果；其中，改进YOLOv5模型包括依此连接的改进YOLOv5神经网络主干部分、YOLOv5神经网络颈部和改进YOLOv5神经网络输出部分；所述改进YOLOv5神经网络主干部分通过在YOLOv5神经网络的主干部分中的C3模块添加注意力机制NAM模块和在YOLOv5神经网络的主干模块添加SPD‑Conv模块获取；所述改进YOLOv5神经网络输出部分通过在YOLOv5神经网络的输出部分之后增加Alpha‑IOU模块获取；通过在YOLOv5主干网络的C3模块增加NAM注意力机制提高了模型对小目标缺陷的提取能力，同时引入新的CNN模块SPD‑Conv，来解决布匹瑕疵较小时，性能迅速下降的问题。