Collaborative Global-Local Networks for Memory-Efficient Segmentation of Ultra-High Resolution Images

被引:63
作者
Chen, Wuyang [1 ]
Jiang, Ziyu
Wang, Zhangyang
Cui, Kexin
Qian, Xiaoning
机构
[1] Texas A&M Univ, Dept Comp Sci & Engn, College Stn, TX 77843 USA
来源
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019) | 2019年
基金
美国国家科学基金会;
关键词
D O I
10.1109/CVPR.2019.00913
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Segmentation of ultra-high resolution images is increasingly demanded, yet poses significant challenges for algorithm efficiency, in particular considering the (GPU) memory limits. Current approaches either downsample an ultrahigh resolution image or crop it into small patches for separate processing. In either way, the loss of local fine details or global contextual information results in limited segmentation accuracy. We propose collaborative Global-Local Networks (GLNet) to effectively preserve both global and local information in a highly memory-efficient manner. GLNet is composed of a global branch and a local branch, taking the downsampled entire image and its cropped local patches as respective inputs. For segmentation, GLNet deeply fuses feature maps from two branches, capturing both the high-resolution fine structures from zoomed-in local patches and the contextual dependency from the down sampled input. To further resolve the potential class imbalance problem between background and foreground regions, we present a coarse-to-fine variant of GLNet, also being memory-efficient. Extensive experiments and analyses have been performed on three real-world ultra-high aerial and medical image datasets (resolution up to 30 million pixels). With only one single 1080Ti GPU and less than 2GB memory used, our GLNet yields high-quality segmentation results and achieves much more competitive accuracy-memory usage trade-offs compared to state-of-the-arts.
引用
收藏
页码:8916 / 8925
页数:10
相关论文
共 43 条
  • [1] [Anonymous], 2018, PYTORCH IMPLEMENTATI
  • [2] [Anonymous], PROC CVPR IEEE
  • [3] [Anonymous], 2018, IEEE INT SYMP CIRC S, DOI DOI 10.1109/ISCAS.2018.8351436
  • [4] [Anonymous], 2017, P IEEE C COMP VIS PA
  • [5] [Anonymous], DIG CIN SYST SPEC VE
  • [6] [Anonymous], 2015, PROC CVPR IEEE
  • [7] [Anonymous], 2018, P BRIT MACH VIS C
  • [8] [Anonymous], 2018, ARXIV180901826
  • [9] [Anonymous], 2017, IEEE INT GEOSC REM S
  • [10] [Anonymous], 2017, IEEE T PATTERN ANAL, DOI DOI 10.1109/TPAMI.2016.2644615