KM_GrabCut - A Fast Interactive Image Segmentation Algorithm

摘要

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.