一种基于改进YOLOv5l网络的木材表面缺陷检测方法

    公开(公告)号:CN114359235B

    公开(公告)日:2025-04-04

    申请号:CN202210015819.X

    申请日:2022-01-07

    Abstract: 本发明涉及一种基于改进YOLOv5l网络的木材表面缺陷检测方法,该方法包括:获取公开的木材缺陷数据集作为数据集一。构建改进的YOLOv5l网络模型,利用数据集一对改进的YOLOv5l网络模型进行训练,得到训练好的网络模型model1。采集实景木材数据作为数据集二,数据集二分为训练集、验证集和测试集。采用model1对测试集进行预测,将测试集中带有缺陷的木材图像标记上标签,将大于设定预测阈值的标签作为伪标签。将数据集二中的训练集和测试集微调后来训练model1,得到model2,采用验证集在model1与model2中选择出最佳的网络模型。利用最佳网络模型对木材表面缺陷进行检测。本发明能够对各种类别的木材表面缺陷进行检测,有效提高了识别准确率和识别效率。

Patent Agency Ranking