基于空洞可分离卷积的编码器-解码器用于语义图像分割

Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentation

陈良杰 Liang-Chieh Chen · Google · 2018-02-07 · arXiv:1802.02611 ↗ · 被引 16970

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

深度神经网络在语义分割任务中使用了空间金字塔池化模块或编码器-解码器结构。前者通过以多种速率和有效视野对输入特征进行滤波或池化操作,能够编码多尺度上下文信息;而后者通过逐步恢复空间信息,能够捕获更锐利的物体边界。在这项工作中,我们提出结合两种方法的优势。具体来说,我们提出的模型 DeepLabv3+通过添加一个简单而有效的解码器模块来细化分割结果,尤其是沿着物体边界。我们进一步探索了 Xception 模型,并将深度可分离卷积应用于空洞空间金字塔池化和解码器模块,从而得到一个更快更强的编码器-解码器网络。我们在 PASCAL VOC 2012 和 Cityscapes 数据集上展示了所提模型的有效性,在无需任何后处理的情况下,测试集性能分别达到 89.0%和 82.1%。我们的论文附带了所提模型在 Tensorflow 中的公开参考实现,网址为 https://github.com/tensorflow/models/tree/master/research/deeplab。

Spatial pyramid pooling module or encode-decoder structure are used in deep neural networks for semantic segmentation task. The former networks are able to encode multi-scale contextual information by probing the incoming features with filters or pooling operations at multiple rates and multiple effective fields-of-view, while the latter networks can capture sharper object boundaries by gradually recovering the spatial information. In this work, we propose to combine the advantages from both methods. Specifically, our proposed model, DeepLabv3+, extends DeepLabv3 by adding a simple yet effective decoder module to refine the segmentation results especially along object boundaries. We further explore the Xception model and apply the depthwise separable convolution to both Atrous Spatial Pyramid Pooling and decoder modules, resulting in a faster and stronger encoder-decoder network. We demonstrate the effectiveness of the proposed model on PASCAL VOC 2012 and Cityscapes datasets, achieving the test set performance of 89.0\% and 82.1\% without any post-processing. Our paper is accompanied with a publicly available reference implementation of the proposed models in Tensorflow at \url{https://github.com/tensorflow/models/tree/master/research/deeplab}.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 13)

阅读逐段中英对照全文 →