[发明专利]一种基于Mask Scoring R-CNN网络的受电弓检测方法

权利要求书

1.一种基于Mask Scoring R-CNN网络的受电弓检测方法，其特征在于，该方法包括以下步骤：

S1、采集弓网红外受电弓数据并进行数据预处理，将数据划分为训练样本集和测试样本集，构建受电弓目标数据库；

S2、搭建受电弓检测网络，利用主干网络提取受电弓多尺度特征图，通过预测头和掩码头获得受电弓的类别信息、位置坐标和粗粒度分割结果，并提出一种边缘修复方法对受电弓粗粒度分割结果进行精细化修复；

S3、将训练样本集的数据加载至受电弓检测网络中，反复迭代训练，并进行参数调整，获得优质受电弓检测模型；

S4、加载优质受电弓检测模型，将测试样本集的数据输入模型，评价受电弓检测分割结果。

2.根据权利要求1所述的基于Mask Scoring R-CNN网络的受电弓检测方法，其特征在于，所述步骤S1具体包括：

S11、通过红外相机采集弓网红外受电弓数据，并对数据进行解码和场景分类；

S12、对数据进行预处理，并分配训练样本集和测试样本集，构建目标数据库。

3.根据权利要求2所述的基于Mask Scoring R-CNN网络的受电弓检测方法，其特征在于，所述步骤S11具体包括：

通过安装在受电弓前的车载接触网运行状态检测装置中的红外相机进行弓网红外视频素材拍摄，并将视频进行解码，获得图像10000帧；采集到的弓网红外受电弓图像的场景包括山体、桥梁、隧道、铁架桥，天气状况包括和阴雨、大雾、阳光照射、大雪，满足受电弓检测数据的多样性，并将所有图片保存至指定路径的弓网红外图像文件夹。

4.根据权利要求3所述的基于Mask Scoring R-CNN网络的受电弓检测方法，其特征在于，所述步骤S12具体包括：

统计10000张弓网红外图像中受电弓标定框的四个坐标点横纵坐标，并对横纵坐标值分别取并集，构成的最大区域即为受电弓所在范围；为防止受电弓在行驶过程中可能出现上下起伏的现象并提升鲁棒性，将受电弓所在范围对应的标定框上下边界扩大一定的像素值，并分别从原图像中裁剪出固定尺寸的子图像，进而构建受电弓模板数据库；利用深度学习开源标注工具LabelMe对受电弓模板数据库中的10000张图像进行标注，分配6000张图像作为训练集，其余4000张图像作为测试集，且训练数据和测试数据没有重复图像，进而成功构建受电弓目标数据库。

5.根据权利要求1所述的基于Mask Scoring R-CNN网络的受电弓检测方法，其特征在于，所述步骤S2建立受电弓检测网络具体包括：

S21、主干网络Backbone使用残差网络ResNet101和特征金字塔网FPN提取受电弓多尺度特征图；

S22、运用区域推荐网络RPN来提议候选窗口区域Proposals；检测头RCNN Head对RPN产生的每个候选区域进行感兴趣区域对齐RoIAlign操作；

S23、针对RoIAlign后的特征图，通过一系列卷积层、全连接层对候选区域的受电弓目标进行分类和边界框回归操作；

掩码头Mask Head用于获得实例像素级的特征，经过6层卷积操作实现特征图的上采样和通道维度的改变，并生成对应的预测掩码；

掩码交并比头MaskIoU Head使用预测掩码的输出和感兴趣区域对齐后RoIAlign的特征作为输入，经过4个卷积层实现特征图的降采样，并连接3个全连接层，最后一层输出C个类别的MaskIoU得分，并与检测头RCNN Head中的类别分数相乘获得更准确的掩码得分；

S24、掩码边缘修复头对掩码头Mask Head获得的粗粒度掩码与真值掩码计算交并比，匹配训练集中最佳的受电弓边界，并获得受电弓边界精细化修复结果。