发明公开
- 专利标题: 一种基于改进YOLOv5的刨花板表面缺陷检测方法和装置
-
申请号: CN202310922228.5申请日: 2023-07-25
-
公开(公告)号: CN117058077A公开(公告)日: 2023-11-14
- 发明人: 关淯尹 , 王文财 , 苍志智 , 查健 , 沈彤 , 李嘉禄 , 党亚光 , 朱翔
- 申请人: 北京建筑材料科学研究总院有限公司 , 北京金隅集团股份有限公司
- 申请人地址: 北京市石景山区金顶北路69号;
- 专利权人: 北京建筑材料科学研究总院有限公司,北京金隅集团股份有限公司
- 当前专利权人: 北京建筑材料科学研究总院有限公司,北京金隅集团股份有限公司
- 当前专利权人地址: 北京市石景山区金顶北路69号;
- 代理机构: 北京路浩知识产权代理有限公司
- 代理商 王婷
- 主分类号: G06T7/00
- IPC分类号: G06T7/00 ; G06T7/62 ; G06V10/25 ; G06V10/774 ; G06V10/80 ; G06V10/82 ; G06N3/045
摘要:
本发明提供一种基于改进YOLOv5的刨花板表面缺陷检测方法和装置,方法包括:获取包含预定比例小尺寸缺陷的刨花板表面缺陷图,构建刨花板表面小尺寸缺陷数据集;基于YOLOv5网络,在主干网络中采用MobileViT模块替代最末端的C3模块,将主干网络的P2层输出的特征信息加入所述特征融合网络,在检测模块中增加CA注意力机制,并引入Focal Loss损失函数,构建针对小尺寸缺陷检测的改进YOLOv5刨花板缺陷检测模型;基于所述刨花板表面小尺寸缺陷数据集,对所述模型进行训练;利用训练后的模型对刨花板表面缺陷进行检测。本发明基于YOLOv5s网络,构建针对刨花板表面小尺寸缺陷的检测网络,使网络更关注于小尺寸缺陷目标样本,提升网络对于小尺寸缺陷的检测效果。