• Laser & Optoelectronics Progress
  • Vol. 58, Issue 2, 0210024 (2021)
Cuijun Zhang1、2 and Na Zhao1、*
Author Affiliations
  • 1School of Information Engineering, Hebei GEO University, Shijiazhuang, Hebei 0 50031, China
  • 2Hebei Center for Ecological and Environmental Geology Research, Hebei GEO University, Shijiazhuang, Hebei 0 50031, China
  • show less
    DOI: 10.3788/LOP202158.0210024 Cite this Article Set citation alerts
    Cuijun Zhang, Na Zhao. Improved GrabCut Algorithm Based on Probabilistic Neural Network[J]. Laser & Optoelectronics Progress, 2021, 58(2): 0210024 Copy Citation Text show less

    Abstract

    Aiming at the low efficiency of GrabCut algorithm in image segmentation, and the problems of under-segmentation and over-segmentation, an improved GrabCut algorithm based on probabilistic neural network (PNN) (PNN_GrabCut) is proposed in this paper. The algorithm replaces the Gaussian mixture model (GMM) in the GrabCut algorithm with PNN model to calculate the weight of t-links to improve the calculation efficiency of the algorithm. By constructing the foreground and background histograms, the pixels with higher pixel values are selected as training samples of the PNN model to improve the segmentation accuracy of the algorithm. In the public ADE20K data set, images are selected for segmentation experiments. The results show that the segmentation accuracy of PNN_GrabCut algorithm is better than other comparison algorithms, and the efficiency is higher. For image segmentation experiments with high similarity between foreground and background, the results show that the segmentation accuracy of PNN_GrabCut algorithm is significantly higher than that of GrabCut algorithm.
    Cuijun Zhang, Na Zhao. Improved GrabCut Algorithm Based on Probabilistic Neural Network[J]. Laser & Optoelectronics Progress, 2021, 58(2): 0210024
    Download Citation