Small Object Sensitive Segmentation of Urban Street Scene With Spatial Adjacency Between Object Classes

被引:48
作者
Guo, Dazhou [1 ]
Zhu, Ligeng [2 ]
Lu, Yuhang [1 ]
Yu, Hongkai [3 ]
Wang, Song [1 ,4 ]
机构
[1] Univ South Carolina, Dept Comp Sci & Engn, Columbia, SC 29201 USA
[2] MIT, Dept Elect Engn & Comp Sci, Cambridge, MA 02319 USA
[3] Univ Texas Rio Grande Valley, Dept Comp Sci, Edinburg, TX 78539 USA
[4] Tianjin Univ, Sch Comp Sci & Technol, Tianjin 300072, Peoples R China
基金
美国国家科学基金会;
关键词
Small objects segmentation; spatial adjacency; semantic segmentation; urban street scene; LEVEL SET METHOD;
D O I
10.1109/TIP.2018.2888701
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Recent advancements in deep learning have shown an exciting promise in the urban street scene segmentation. However, many objects, such as poles and sign symbols, are relatively small, and they usually cannot be accurately segmented, since the larger objects usually contribute more to the segmentation loss. In this paper, we propose a new boundary-based metric that measures the level of spatial adjacency between each pair of object classes and find that this metric is robust against object size-induced biases. We develop a new method to enforce this metric into the segmentation loss. We propose a network, which starts with a segmentation network, followed by a new encoder to compute the proposed boundary-based metric, and then trains this network in an end-to-end fashion. In deployment, we only use the trained segmentation network, without the encoder, to segment new unseen images. Experimentally, we evaluate the proposed method using CamVid and CityScapes data sets and achieve a favorable overall performance improvement and a substantial improvement in segmenting small objects.
引用
收藏
页码:2643 / 2653
页数:11
相关论文
共 46 条
[1]  
[Anonymous], P SPIE
[2]  
[Anonymous], P 3 INT C LEARNING R
[3]  
[Anonymous], PROC CVPR IEEE
[4]  
[Anonymous], 2017, COMMUN ACM, DOI DOI 10.1145/3065386
[5]  
[Anonymous], 2015, PROC CVPR IEEE
[6]  
[Anonymous], 2017, IEEE I CONF COMP VIS, DOI DOI 10.1109/ICCV.2017.147
[7]  
[Anonymous], 2017, C COMP VIS PATT REC
[8]  
[Anonymous], MERGENET DEEP NET AR
[9]  
[Anonymous], 2017, IEEE T PATTERN ANAL, DOI DOI 10.1109/TPAMI.2016.2644615
[10]  
[Anonymous], 2016, DEEPLAB SEMANTIC IMA