一种基于改进YOLOX的竹条表面缺陷检测方法

    公开(公告)号:CN117078614A

    公开(公告)日:2023-11-17

    申请号:CN202311006374.X

    申请日:2023-08-11

    Abstract: 本发明提供了一种基于改进YOLOX的竹条表面缺陷检测方法,包括以下步骤:步骤1:采集竹条表面缺陷数据,并进行标注,然后按7:2:1的比例将数据集划分成训练集、验证集和测试集;步骤2:构建改进的YOLOX网络;步骤3:使用构建的数据集训练改进的YOLOX缺陷检测网络,训练前先对训练的图像使用HSV色调变换和Mosaic数据增强方法对进行数据增强操作;损失函数包含分类损失、置信度损失和边界框回归损失;其中将置信度损失修改为Focal Loss,以改善正负样本不均衡的问题,将边界框回归损失修改为CIoU Loss;步骤4:将待检测的图像输入训练好的网络中,得到缺陷的类别和位置信息。应用本技术方案可实现自动对缺陷进行检测,同时具有较高的准确率。

Patent Agency Ranking