Remote sensing image semantic segmentation combining UNET and FPN

被引:10
|
作者
Wang Xi [1 ]
Yu Ming [1 ,2 ]
Ren Hong-e [1 ,2 ]
机构
[1] Northeast Forestry Univ, Coll Informat & Comp Engn, Harbin 150040, Peoples R China
[2] Heilongjiang Forestry Intelligent Equipment Engn, Harbin 150040, Peoples R China
关键词
deep learning; UNET; FPN; BLR;
D O I
10.37188/CJLCD.2020-0116
中图分类号
O7 [晶体学];
学科分类号
0702 ; 070205 ; 0703 ; 080501 ;
摘要
The traditional remote sensing image segmentation method is inefficient and the segmentation fineness is not enough in complex scenes. The UNET model is well-known for its good segmentation effect, but it does not perform well for the smaller objects contained in the image and the edge segmentation of larger objects. In order to solve this problem, a method combining UNET structure with FPN structure is proposed in this paper to improve the ability of UNET model to integrate multi-scale information. At the same time, the BLR loss function which can better capture the edge of the target edge is used to improve the segmentation effect of UNET model on the target boundary. The experimental results show that the method used in this paper effectively improves the accuracy of semantic segmentation and alleviates the problem of poor edge segmentation of small-scale targets and large-scale targets. The target edge segmentation can be more accurate to achieve better segmentation results.
引用
收藏
页码:475 / 483
页数:9
相关论文
共 19 条
  • [1] SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentation
    Badrinarayanan, Vijay
    Kendall, Alex
    Cipolla, Roberto
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2017, 39 (12) : 2481 - 2495
  • [2] Cascade R-CNN: High Quality Object Detection and Instance Segmentation
    Cai, Zhaowei
    Vasconcelos, Nuno
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2021, 43 (05) : 1483 - 1498
  • [3] Cao Yuliang, 2017, Journal of Xi'an Jiaotong University, V51, P165, DOI 10.7652/xjtuxb201711023
  • [4] DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFs
    Chen, Liang-Chieh
    Papandreou, George
    Kokkinos, Iasonas
    Murphy, Kevin
    Yuille, Alan L.
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2018, 40 (04) : 834 - 848
  • [5] CHEN XL, 2019, P IEEE CVF INT C COM
  • [6] [陈彦彤 Chen Yantong], 2020, [光学精密工程, Optics and Precision Engineering], V28, P1165
  • [7] Multi-view Face Detection Using Deep Convolutional Neural Networks
    Farfade, Sachin Sudhakar
    Saberian, Mohammad
    Li, Li-Jia
    [J]. ICMR'15: PROCEEDINGS OF THE 2015 ACM INTERNATIONAL CONFERENCE ON MULTIMEDIA RETRIEVAL, 2015, : 643 - 650
  • [8] Feiyan Zhou, 2017, J COMPUTER SCI, V40, P1229, DOI DOI 10.11897/SP.J.1016.2017.01229
  • [9] KOKKINOS I, 2015, P INT C LEARN REPR U
  • [10] ImageNet Classification with Deep Convolutional Neural Networks
    Krizhevsky, Alex
    Sutskever, Ilya
    Hinton, Geoffrey E.
    [J]. COMMUNICATIONS OF THE ACM, 2017, 60 (06) : 84 - 90