一种基于改进YOLOv5的刨花板表面缺陷检测方法和装置
摘要:
本发明提供一种基于改进YOLOv5的刨花板表面缺陷检测方法和装置,方法包括:获取包含预定比例小尺寸缺陷的刨花板表面缺陷图,构建刨花板表面小尺寸缺陷数据集;基于YOLOv5网络,在主干网络中采用MobileViT模块替代最末端的C3模块,将主干网络的P2层输出的特征信息加入所述特征融合网络,在检测模块中增加CA注意力机制,并引入Focal Loss损失函数,构建针对小尺寸缺陷检测的改进YOLOv5刨花板缺陷检测模型;基于所述刨花板表面小尺寸缺陷数据集,对所述模型进行训练;利用训练后的模型对刨花板表面缺陷进行检测。本发明基于YOLOv5s网络,构建针对刨花板表面小尺寸缺陷的检测网络,使网络更关注于小尺寸缺陷目标样本,提升网络对于小尺寸缺陷的检测效果。
0/0