Laser & Optoelectronics Progress, Vol. 58, Issue 2, 0210024 (2021)
Improved GrabCut Algorithm Based on Probabilistic Neural Network
Zhang Cuijun1,2 and Zhao Na1,*
- 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
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.
Please Enter Your Email: