[发明专利]交通场景分割方法及相关装置

说明书

本申请实施例公开了一种交通场景分割方法及相关装置，方法包括：将原始图像输入到主干网络，得到融合特征图；将融合特征图输入到分割网络，得到分割特征图；将融合特征图输入到形状网络，得到第一形状特征图，优化第一形状特征图，得到第二形状特征图；将分割特征图和第二形状特征图合并得到合并特征网络；将合并特征网络进行第一卷积运算，得到中间运算结果；将其进行解码操作，得到第一显著性特征图，对其进行优化处理，得到第二显著性特征图；依据第二显著性特征图确定带方向建议框；将带方向建议框依据进行第二卷积运算和ASPP运算，得到得分特征图；优化得分特征图，得到分割结果图像。采用本申请实施例能够提升交通场景分割精度。

技术领域

本申请涉及图像处理技术领域，具体涉及一种交通场景分割方法及相关装置。

背景技术

随着社会的发展，城市汽车数量不断攀升，由此带来的交通和环境问题也日渐加剧。为了应对这些问题，智能交通系统已成为城市发展的重点研究对象。而在智能交通系统中，对于车辆的检索，已经被证明是一项关键技术，在处理包括套牌车、故意遮挡车牌和追踪肇事逃逸等等交通问题上有着得天独厚的优势，对于智能交通系统的建设具有重要的意义，因此，如何提升交通场景分割精度的问题亟待解决。

发明内容

本申请实施例提供了一种交通场景分割方法及相关装置，可以提升行交通场景分割精度。

第一方面，本申请实施例提供一种交通场景分割方法，应用于电子设备，所述方法包括：

获取原始图像，以及所述原始图像的形状标记图和点类别标记图，所述原始图像为包括目标的图像，所述形状标记图为所述目标的边缘轮廓图像，所述点类别标记图为所述目标的区域图像；

将所述原始图像输入到主干网络，得到融合特征图；

将所述融合特征图输入到分割网络，得到分割特征图；

将所述融合特征图输入到形状网络，得到第一形状特征图，并通过所述形状标记图优化所述第一形状特征图，得到第二形状特征图；

将所述分割特征图和所述第二形状特征图进行合并，得到合并特征网络；

将所述合并特征网络进行第一卷积运算，得到中间运算结果；

将所述中间运算结果进行解码操作，得到第一显著性特征图，并通过所述第一显著性特征图对应的二值图对所述第一显著性特征图进行优化处理，得到第二显著性特征图；

依据所述第二显著性特征图确定带方向建议框；

将所述带方向建议框依据进行第二卷积运算和ASPP运算，得到得分特征图；

依据所述点类别标记图优化所述得分特征图，得到分割结果图像。

第二方面，本申请实施例提供了一种交通场景分割装置，应用于电子设备，所述装置包括：获取单元、输入单元、合并单元、运算单元、解码单元、确定单元和优化单元，其中，

所述获取单元，用于获取原始图像，以及所述原始图像的形状标记图和点类别标记图，所述原始图像为包括目标的图像，所述形状标记图为所述目标的边缘轮廓图像，所述点类别标记图为所述目标的区域图像；

所述输入单元，用于将所述原始图像输入到主干网络，得到融合特征图；以及将所述融合特征图输入到分割网络，得到分割特征图；将所述融合特征图输入到形状网络，得到第一形状特征图，并通过所述形状标记图优化所述第一形状特征图，得到第二形状特征图；

所述合并单元，用于将所述分割特征图和所述第二形状特征图进行合并，得到合并特征网络；

所述运算单元，用于将所述合并特征网络进行第一卷积运算，得到中间运算结果；