[发明专利]一种基于改进EfficientDet模型的织物瑕疵检测方法

权利要求书

1.一种基于改进EfficientDet模型的织物瑕疵检测方法，其特征在于：包括如下工艺步骤：

1)，使用工业相机采集织物数据集，对图像疵点进行标注，以获得瑕疵类别标签；

2)，对织物数据集和对应的类别标签进行数据增强，满足网络模型训练的样本数量要求；

3)，将样本增强数据集按预设比例划分为训练集、验证集、测试集；

4)，载入预训练模型，预设好模型训练的Batch_size，任意选取训练集中一个批次张图片输入至模型中；选取每张图片在主干网络EfficientNet中N个特定阶层的有效特征图，经过通道变换后传入到特征金字塔网络进行多尺度信息整合得到N个对应阶层的加强特征图；

5)，每个加强特征图均独立地输入到分类/回归网络并进行候选框的调整；

6)，网络根据加强特征图图、调整后的候选框与相应真实框差异计算出分类和回归损失，并根据网络整体损失对整个网络模型参数进行更新；

7)，通过参数更新后的改进EfficientDet模型在验证集上计算验证集损失，根据验证集损失曲线考量该模型的泛化性能；

8)，经过预先设立的P轮Epoch训练迭代，即重复P次步骤4)～7)后，网络总损失曲线收敛，训练结束，得到最终的改进EfficientDet网络模型；

9)，载入步骤8)得到的训练模型，遍历测试集文件夹中所有图片进行预测，依据非极大值抑制原则进行候选框筛选，然后从所有调整后的测试集候选框中根据置信度比较得到最终预测框，实现瑕疵的定位与分类。

2.如权利要求1所述的基于改进EfficientDet模型的织物瑕疵检测方法，其特征在于：所述步骤1)中，使用LabelImg软件中最小外接矩形框对每张照片疵点进行分类别标定，得到类别标签XML文件；所述XML文件记录的信息包括图像的尺寸、瑕疵的类别、边界框的空间坐标位置等。

3.如权利要求1所述的基于改进EfficientDet模型的织物瑕疵检测方法，其特征在于：所述步骤2)中，数据增强手段包括图像翻转、旋转、裁剪、亮度变化、对比度变化、增加噪声。

4.如权利要求1所述的基于改进EfficientDet模型的织物瑕疵检测方法，其特征在于：所述步骤3)中，(训练集+验证集)：测试集的样本比例为9：1；其中训练集：验证集的样本比例为9：1；此时会生成train.txt、val.txt、test.txt文件来记录各自的划分后的图像名称列表；然后根据train.txt、val.txt生成2007_train.txt、2007_val.txt。

5.如权利要求1所述的基于改进EfficientDet模型的织物瑕疵检测方法，其特征在于：所述步骤4)具体包括如下过程：

4-1)，以EfficientDet-d0作为基础模型，并载入该模型预训练权重；每张图片首先输入至主干网络进行初步的特征提取，得到5个不同阶层(P3、P4、P5、P6、P7)的有效特征图；所述主干网络由不同层数的MBConv结构块和末端的并行膨胀注意力模块PDAM堆叠而成；其中每一阶层MBConv结构块代表主干网络的一个Stage；

4-2)，将主干网络提取到的5个不同阶层(P3、P4、P5、P6、P7)的特征图进行卷积操作实现通道统一后，输入到B-BiFPN特征金字塔网络中，不同网络层之间以上/下采样的方式进行特征融合，得到5个不同尺度(分辨率分别为64×64、32×32、16×16、8×8、4×4)的精细化的加强特征图。