RGPNET: 复杂环境下实时通用语义分割网络

原文链接:https://arxiv.org/abs/1912.01394

一、主要思想

本文提出了一种新的实时通用语义分割体系结构RGPNet,在复杂环境下取得了显著的性能提升。RGPNet由一个轻量级的非对称编码器-解码器和一个适配器组成。适配器有助于从编码器和解码器之间的多层分布式表示中保留和细化抽象概念。它也有助于从较深层到较浅层的梯度流动。大量实验表明,与目前最先进的语义分割网络相比,RGPNet具有更好的性能。

此外还证明了在保持性能的同时,使用改进的标签松弛技术和逐步调整大小可以减少60%的训练时间。论文还对应用在资源受限的嵌入式设备上的RGPNet进行了优化,使推理速度提高了400%,性能损失可以忽略不计。RGPNet在多个数据集之间获得了更好的速度和精度权衡。

二、创新点

1、提出的RGPNet作为一种通用的实时语义分割体系结构,它可以在单分支网络中获得高分辨率的深层特征,从而提高准确性和降低延迟,在复杂的环境中表现突出。

2、引入一个适配器模块来捕获多个抽象级别,以帮助细分的边界细化。

3、在训练期间采用渐进式调整大小技术,从而使训练时间和环境影响减少60%,并且采用一种改进的标签松弛来消除低分辨率标签映射中的混叠效应。

4、使用TensorRT优化RGPNet,以便部署在边缘计算设备上,从而使推理速度提高400%。

5、RGPNet在Cityscpes、CamVid和Mapillary数据集上分别实现了Resnet-101作为backbone 下80.9%、69.2%和50.2% mIoU以及Resnet-18作为backbone下74.1%、66.9%和41.7% mIoU。对于1024×2048分辨率的图像,RGPNet在CityScapes数据集上单NVIDIA GTX2080Ti GPU下达到37.4 FPS。

三、网络结构

RGPNet的整体结构如下所示:

 

  • RGPNET: 复杂环境下实时通用语义分割网络

  • 其中“+”详细操作如下所示:

    1、T(:)是一个转换函数,它用来减少编码器模块输出通道数量并将其传输到adaptor。

    2、D(:)和U(:)是下采样和上采样功能。

  • RGPNET: 复杂环境下实时通用语义分割网络

    • Adaptor有许多优点:

      1、Adaptor聚合来自不同上下文和空间级别的特征。

      2、通过引入较短的路径,有助于梯度从较深的层流向较浅的层。

      3、Adaptor允许使用轻量解码器的不对称设计,这将减少卷积层,进一步增强梯度流。因此,Adaptor使网络适合于实时应用,因为它在保留空间信息的同时提供了丰富的语义信息。

      针对带标签松弛的渐进式调整:

      论文采取了最大化像素周围区域相似度分布,而不是单个像素级别的标签最大可能化,针对边界类别,提出了边界损失函数。

      四、实验结果

      多种网络在Mapillary Vistas数据集上的测试结果:

      RGPNET: 复杂环境下实时通用语义分割网络

       

      RGPNET: 复杂环境下实时通用语义分割网络

       

       

       

      在CamVid数据集上的性能对比:

      RGPNET: 复杂环境下实时通用语义分割网络

      思考:RGPNet在cityscapes数据集上用resnet101可以达到80.9%的准确率,而且可以在嵌入式平台进行优化,比较适合目前进行的工作。但是resnet101参数计算量都比较大,可以尝试用小的主干网络代替。。