KM_GrabCut: a fast interactive image segmentation algorithm
Jianbo Li, Yiping Yao, Wenjie Tang · Proceedings of SPIE, the International Society for Optical Engineering/Proceedings of SPIE · 2015
Image segmentation is critical for image processing. Among several algorithms, GrabCut is well known by its little user interaction and desirable segmentation result. However, it needs to take a lot of time to adjust the Gaussian Mixture Model (GMM) and to cut the weighted graph with Max-Flow/Min-Cut Algorithm iteratively. To solve this problem, we first build a common algorithmic framework which can be shared by the class of GrabCut-like segmentation algorithms, and then propose KM_GrabCut algorithm based on this framework. The KM_GrabCut first uses K-means clustering algorithm to cluster pixels in foreground and background respectively, and then constructs a GMM based on each clustering result and cuts the corresponding weighted graph only once. Experimental results demonstrate that KM_GrabCut outperforms GrabCut with higher performance, comparable segmentation result and user interaction.