[发明专利]一种变电站头盔和工作服检测方法及装置

权利要求书

1.一种变电站头盔和工作服检测方法，其特征在于，所述方法包括：

采集变电站工作人员巡检图像，在图像中标记头盔和工作服，作为第一标签；

将标记后的图像作为YOLO3模型的输入，将输出结果框编号组成结果序列并保存在容器A中；

从容器A中分别筛选不同目标对应的结果框及对应的结果序列；

对于容器A中重复结果框部分进行结果框重置计算、标签重新匹配和置信度重新匹配，其他部分保持不变，得到新的结果序列，保存容器C中；

以容器A中的原始序列为输入序列，以容器C中的新的结果序列为目标序列，训练seq2seq模型，得到结果框转化模型；

将YOLO3模型与结果框转化模型进行串联整合，得到融合模型；

将待识别图像输入所述融合模型，删除输出的结果序列中置信度小于预设阈值的序列，得到无重复框的检测结果。

2.根据权利要求1所述变电站头盔和工作服检测方法，其特征在于，所述结果序列中包括序号，置信度，类别，位置，所述位置包括结果框的中心坐标、宽度、高度。

3.根据权利要求1所述变电站头盔和工作服检测方法，其特征在于，所述分别筛选不同目标对应的结果框具体包括：

预先设定交并比阈值T；设定同一目标不同结果框中心坐标之间的距离阈值D；将所有结果框按置信度降序排列；

从降序排列结果中取出置信度最高的结果框，分别计算置信度最高的结果框与其余所有结果框之间的坐标中心距离和交并比，筛选出与置信度最高的结果框之间类别相同且坐标中心距离小于D且交并比大于T的结果框，将置信度最高的结果框与筛选出的结果框作为同一目标的结果框；将同一目标的结果框从所述降序排列结果中剔除；

重复上述从降序排列结果中同一目标的结果框过程，直到确定所有目标对应的结果框。

4.根据权利要求2所述变电站头盔和工作服检测方法，其特征在于，所述对于重复结果框部分进行结果框重置计算具体包括：

判断各个目标是否包含重复结果框，将容器A中所有重复结果框对应的结果序列保存在容器B中；

当容器B中同一目标的结果序列中结果框重复时，通过各个结果框的中心坐标之间的距离、交并比、置信度计算当前目标的重置结果框位置；

用所述重置结果框位置替换容器B中对应目标的结果框中与第一标签的交并比最大的结果框的位置，序号、置信度、类别均保持不变。

5.根据权利要求4所述变电站头盔和工作服检测方法，其特征在于，所述通过各个结果框的中心坐标之间的距离、交并比、置信度计算当前目标的重置结果框位置具体包括：

对于第j个目标，设结果框i的中心坐标为(x_i，y_i)，宽为w_i，高为h_i，与第一标签的交并比为ρ_i，置信度为c_i，i＝1,2,…,n，其中交并比最高的结果框对应的中心坐标为(x₁，y₁)，宽为w₁，高为h₁，当前目标j的重置结果框位置计算公式为：

其中，(x_j，y_j)为目标j的重置结果框中心坐标，w_j为重置结果框宽度，h_j为重置结果框高度，c_max＝max{c_i}，ω₁、ω₂为权重系数。