Python+OpenCV：交互式图像前景提取(Interactive Foreground Extraction using GrabCut Algorithm)

素颜马尾好姑娘i 2022-12-23 00:35 58阅读 0赞

## Python+OpenCV：交互式图像前景提取(Interactive Foreground Extraction using GrabCut Algorithm) ##

####################################################################################################
    # 交互式图像前景提取(Interactive Foreground Extraction using GrabCut Algorithm)
    def lmc_cv_image_interactive_foreground_extraction_grabcut():
        """
            函数功能: 交互式图像前景提取(Interactive Foreground Extraction using GrabCut Algorithm)。
        """
    
        # 读取图像
        image = lmc_cv.imread('D:/99-Research/Python/Image/messi.jpg', flags=lmc_cv.IMREAD_UNCHANGED)
        rgb_image = lmc_cv.cvtColor(image, lmc_cv.COLOR_BGR2RGB)
        rgb_image_result = rgb_image.copy()
        gray_image = lmc_cv.cvtColor(rgb_image, lmc_cv.COLOR_RGB2GRAY)
    
        # 基于分水岭算法的图像分割(Image Segmentation with Watershed Algorithm)
        mask = np.zeros(rgb_image.shape[:2], np.uint8)
        bgdModel = np.zeros((1, 65), np.float64)
        fgdModel = np.zeros((1, 65), np.float64)
        rect = (200, 200, 600, 600)
        lmc_cv.grabCut(rgb_image, mask, rect, bgdModel, fgdModel, 5, lmc_cv.GC_INIT_WITH_RECT)
        mask2 = np.where((mask == 2) | (mask == 0), 0, 1).astype('uint8')
        img = rgb_image * mask2[:, :, np.newaxis]
        # 显示图像
        pyplot.figure('Image Display')
        pyplot.imshow(img)
        pyplot.colorbar()
        pyplot.show()
    
        # 根据用户输入保存图像
        if ord("q") == (lmc_cv.waitKey(0) & 0xFF):
            # 销毁窗口
            pyplot.close('all')
        return

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xpdWJpbmc4NjA5_size_16_color_FFFFFF_t_70][]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xpdWJpbmc4NjA5_size_16_color_FFFFFF_t_70]: /images/20221120/7ba68261096c45f989443a22e05c1fa0.png