Semantic image segmentation algorithm in a deep learning computer network

被引:8
作者
He, Defu [1 ]
Xie, Chao [2 ]
机构
[1] Hubei Univ Educ, Coll Comp, Wuhan 430205, Hubei, Peoples R China
[2] Wuhan Donghu Univ, Sch Management, Wuhan 430212, Hubei, Peoples R China
关键词
Semantic image segmentation; Deep learning; Object edges; Difficult examples; Loss function;
D O I
10.1007/s00530-020-00678-1
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Semantic image segmentation in computer networks is designed to determine the category to which each pixel in an image belongs. It is a basic computer vision task and has a very wide range of applications in practice. In recent years, semantic image segmentation algorithms in computer networks based on deep learning have attracted widespread attention due to their fast speed and high accuracy. However, due to the large number of downsampling layers in a deep learning model, the segmentation results are usually poor at the edge of an object, and there is currently no universal quantitative evaluation index to measure the performance of segmentation at the edge of an object. Solving these two problems is of great significance to semantic image segmentation algorithms in China. Based on traditional evaluation indicators, this paper proposes a region-based evaluation index to quantitatively measure the performance of segmentation at the edge of an object and proposes an improved loss function to improve model performance. The existing semantic image segmentation methods are summarized. This paper proposes regional-based evaluation indicators. Taking advantage of the particularity of semantic image segmentation tasks, this paper presents an efficient and accurate method for extracting the edges of objects. By defining the distance from pixels to the edges of objects, this paper proposes a fast algorithm for calculating the edge area. Based on this, three methods are proposed as well as an area-based evaluation indicator. The experimental results show that the accuracy of the loss function proposed in this paper, compared with that of the current mainstream cross-entropy loss function, is improved by 1% on the DeepLab model. For area-based evaluation indicators, a 4% accuracy improvement can be achieved, and on other segmentation models, there is also a significant improvement.
引用
收藏
页码:2065 / 2077
页数:13
相关论文
共 24 条
  • [1] Anupama N., 2017, IET IMAG PROCESS, V11, P777
  • [2] 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
  • [3] Deep learning for visual understanding: A review
    Guo, Yanming
    Liu, Yu
    Oerlemans, Ard
    Lao, Songyang
    Wu, Song
    Lew, Michael S.
    [J]. NEUROCOMPUTING, 2016, 187 : 27 - 48
  • [4] Harley AW, 2015, COMPUT THERM SCI, V79, P1337
  • [5] Level set evolution model for image segmentation based on variable exponent p-Laplace equation
    Huang, Chencheng
    Zeng, Li
    [J]. APPLIED MATHEMATICAL MODELLING, 2016, 40 (17-18) : 7739 - 7750
  • [6] Fast Image-Based Obstacle Detection From Unmanned Surface Vehicles
    Kristan, Matej
    Kenk, Vildana Sulic
    Kovacic, Stanislav
    Pers, Janez
    [J]. IEEE TRANSACTIONS ON CYBERNETICS, 2016, 46 (03) : 641 - 654
  • [7] A Precise Image Crawling System with Image Classification Based on Deep Learning
    Lee, Myung-Jae
    Choi, Suh-Yong
    Jeong, Hyeok-June
    Ha, Young-Guk
    [J]. ADVANCED SCIENCE LETTERS, 2017, 23 (03) : 1623 - 1626
  • [8] Li L, 2017, IEEE T INTELL TRANSP, V99, P1
  • [9] Segmentation of retinal fluid based on deep learning: application of three-dimensional fully convolutional neural networks in optical coherence tomography images
    Li, Meng-Xiao
    Yu, Su-Qin
    Zhang, Wei
    Zhou, Hao
    Xu, Xun
    Qian, Tian-Wei
    Wan, Yong-Jing
    [J]. INTERNATIONAL JOURNAL OF OPHTHALMOLOGY, 2019, 12 (06) : 1012 - 1020
  • [10] [刘李漫 Liu Liman], 2016, [光电子·激光, Journal of Optoelectronics·Laser], V27, P1111