python grabcut算法代码

GrabCut是一种图像分割技术，它结合了用户交互和机器学习。在Python中，我们可以利用OpenCV库来实现GrabCut算法。以下是一个简单的示例代码：

import cv2
import numpy as np

def grabcut(image_path, rect):
    # 加载图像
    img = cv2.imread(image_path)
    
    # 转换为灰度图
    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

    # 定义初始标记区域（背景或前景）
    bgdModel = np.zeros((1,65),np.float64)
    fgdModel = np.zeros((1,65),np.float64)
    rect = (rect[0], rect[1], rect[2], rect[3])
    mask = np.zeros(gray.shape[:2],np.uint8)
    cv2.grabCut(gray,mask,rect,bgdModel,fgdModel,5,cv2.GC_INIT_WITH_RECT)

    # 根据分割结果创建新的掩码
    mask2 = np.where((mask==2)|(mask==0),0,1).astype('uint8')
    img = img*mask2[:,:,np.newaxis]

    return img

# 使用方法
image_path = 'your_image.jpg'
x, y, w, h = 100, 100, 200, 200  # 用户点击的矩形区域坐标
grabcut_output = grabcut(image_path, (x, y, w, h))
cv2.imshow('GrabCut Result', grabcut_output)
cv2.waitKey(0)
cv2.destroyAllWindows()

在这个代码里，`rect`变量表示用户选择的初步分割区域，然后通过`cv2.grabCut`函数进行迭代处理，最后返回的是经过GrabCut处理后的图片。