Xi WANG, Ming YU, Hong-e REN. Remote sensing image semantic segmentation combining UNET and FPN[J]. Chinese journal of liquid crystals and displays, 2021, 36(3): 475-483.
DOI:
Xi WANG, Ming YU, Hong-e REN. Remote sensing image semantic segmentation combining UNET and FPN[J]. Chinese journal of liquid crystals and displays, 2021, 36(3): 475-483. DOI: 10.37188/CJLCD.2020-0116.
Remote sensing image semantic segmentation combining UNET and FPN
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.
关键词
Keywords
references
LECUN Y, BOTTOU L, BENGIO Y, et al . Gradient-based learning applied to document recognition[J]. Proceedings of the IEEE , 1998, 86(11): 2278-2324.
REN S Q, HE K M, GIRSHICK R, et al . Faster R-CNN: towards real-time object detection with region proposal networks[C]// Proceedings of the 28 th International Conference on Neural Information Processing Systems . Montreal, Canada: NIPS, 2015: 91-99.
CAO Y L, MING T F, HE G, et al . Cavitation state recognition of centrifugal pump based on deep learning[J]. Journal of Xi'an Jiaotong University , 2017, 51(11): 165-172. (in Chinese).
FARFADE S S, SABERIAN M J, LI L J. Multi-view face detection using deep convolutional neural networks[C]// Proceedings of the 5 th ACM on International Conference on Multimedia Retrieval . Shanghai: ACM, 2015: 643-650.
KRIZHEVSKY A, SUTSKEVER I, HINTON G E. ImageNet classification with deep convolutional neural networks[C]// Proceedings of the 25 th International Conference on Neural Information Processing Systems . Lake Tahoe, USA: ACM, 2012: 1097-1105.
ZHOU F Y, JIN L P, DONG J. Review of convolutional neural network[J]. Chinese Journal of Computers , 2017, 40(6): 1229-1251. (in Chinese)
SHELHAMER E, LONG J, DARRELL T. Fully convolutional networks for semantic segmentation[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence , 2017, 39(4): 640-651.
LU Y, CHEN Y R, ZHAO D B, et al . Graph-FCN for image semantic segmentation[C]// Proceedings of the 16 th International Symposium on Neural Networks . Moscow, Russia: Springer, 2020: 97-105.
ZHOU R Y, SHEN W Z. PI-Unet: research on a neural network model for accurate segmentation of heterogeneous iris[J/OL]. Computer Engineering and Applications . (2020-08-06)[2020-09-04]. https://www.cnki.net/KCMS/detail/11.2127.TP.20200806.1353.016.html https://www.cnki.net/KCMS/detail/11.2127.TP.20200806.1353.016.html .
LI Z Z, YIN H, ZUO J K, et al . Ship detection method based on UNet++ and multilateral output fusion algorithm[J/OL]. Computer Engineering . (2020-08-07)[2020-09-04]. https://www.cnki.net/KCMS/detail/31.1289.TP.20200806.1722.004.html https://www.cnki.net/KCMS/detail/31.1289.TP.20200806.1722.004.html .
JIANG H D, YE X N. An improved skin disease image segmentation algorithm based on I-Unet network[J]. Modern Electronics Technique , 2019, 42(12): 52-56. (in Chinese)
CHEN X L, GIRSHICK R, HE K M, et al . Tensormask: a foundation for dense object segmentation[C]// Proceedings of the IEEE / CVF International Conference on Computer Vision . Seoul, Korea (South): IEEE, 2019: 2061-2069.
LIANG H, SONG Y L, QIAN F, et al . Detection of small target in aerial photography based on deep learning[J]. Chinese Journal of Liquid Crystals and Displays , 2018, 33(9): 793-800. (in Chinese)
CHEN Y T, LI Y Y, LÜ S L, et al . Research on oil spill monitoring of multi-source remote sensing image based on deep semantic segmentation[J]. Optics and Precision Engineering , 2020, 28(5): 1165-1176. (in Chinese)
ZHU Y, SAPRA K, REDA F A, et al . Improving semantic segmentation via video propagation and label relaxation[C]// Proceedings of the IEEE / CVF Conference on Computer Vision and Pattern Recognition . Long Beach, CA, USA: IEEE, 2019: 8848-8857.
CAI Z W, VASCONCELOS N. Cascade R-CNN: high quality object detection and instance segmentation[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence , 2019, doi:10.1109/TPAMI.2019.2956516.
BADRINARAYANAN V, KENDALL A, CIPOLLA R. SegNet: a deep convolutional encoder-decoder architecture for image segmentation[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence , 2017, 39(12): 2481-2495.
CHEN L C, PAPANDREOU G, KOKKINOS I, et al . Semantic image segmentation with deep convolutional nets and fully connected CRFs[C]// Proceedings of International Conference on Learning Representations . San Diego, United States, 2015.
CHEN L C, PAPANDREOU G, KOKKINOS I, et al . DeepLab: semantic image segmentation with deep convolutional nets, atrous convolution, and fully connected CRFs[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence , 2018, 40(4): 834-848.