[发明专利]一种应用于内窥镜图像目标检测模型的增强训练方法

说明书

本发明公开了一种应用于内窥镜图像目标检测模型的增强训练方法，从样本标签的选择上进行优化，一个是引入对抗样本训练，并优化对抗样本的使用方式，实验表明，这两种策略综合使用，提高了模型在受限数据训练情况下的精度和泛化能力。

技术领域

本发明属于目标检测技术领域，具体的为：一种应用于内窥镜图像目标检测模型的增强训练方法。

背景技术

随着电子技术的发展，医用内窥镜产品越来越丰富，极大的融入了各式各样的疾病诊断、治疗流程中。同时随着图像智能识别技术的发展，利用内窥镜图像，实现部分疾病的自动识别和预警等得到了越来越多的重视。

在图像智能识别中，人工建立的数据库往往具有决定性的作用，但是海量的数据在很多场景下并不存在，尤其是医疗领域，比如妇科内窥镜图像，往往由于各种原因，如隐私等，数据并不能充分的获取和标注。这样就要求对于图像目标检测模型的训练方式，进行尽量的优化。

发明内容

发明目的：针对上述现有技术，提出一种应用于内窥镜图像目标检测模型的增强训练方法

技术方案：一种应用于内窥镜图像目标检测模型的增强训练方法，其特征在于，包括以下步骤：

1)获取待训练图像，所述待训练图像包括标注信息；

2)通过目标检测模型对所述待训练图像进行特征提取，得到所述待训练图像的多尺度特征图；

3)通过所述目标检测模型的特征金字塔网络对所述待训练图像的多尺度特征图进行融合得到融合后的多尺度特征图；

4)通过所述目标检测模型的检测头网络对融合后的多尺度特征图进行检测得到待训练图像的目标检测结果；

5)引入最优对抗样本对所述目标检测模型进行训练，预设锚点集合，通过预设锚点的样本的权重对所述目标检测模型的损失进行调制，得到训练后的目标检测模型。

优选的，预设锚框样本集合为{Q_i}，目标真实位置为R，所述目标检测模型使用所述锚框样本集合{Q_i}得到的预测置信度集合为目标预测位置集合为其中i表示锚点序号；则训练中选取的正锚框样本的权重w_i为：

对所有锚框样本进行权重计算，从大到小排序，得到集合，根据下式计算k值：

其中，T表示固定阈值，选出权重大于固定阈值T的k个正锚框样本。

优选的，通过最大化交叉熵法来获得最优对抗样本，将所述最优对抗样本记为X^a，计算最优对抗样本X^a的均值μ^a和方差σ^a，然后归一化所有的X^a得到将作为所述目标检测模型的batchnorm层的输出。

优选的，所述目标检测模型为YOLOv6。

有益效果：

1)从两个方面，对目标检测训练流程进行了优化。一个是从样本标签的选择上进行优化，一个是引入对抗样本训练，并优化对抗样本的使用方式。实验表面，这两种策略综合使用，提高了模型在受限数据训练情况下的精度和泛化能力。

2)不但维护了网络对于常规样本的响应能力，同时也强化了其对于攻击性样本的鲁棒性

具体实施方式

下面对本发明做更进一步的解释。

一种应用于内窥镜图像目标检测模型的增强训练方法，其特征在于，包括以下步骤：

1)获取待训练图像，所述待训练图像包括标注信息；