Grabcut初始化图像分割

import cv2
import numpy as np
import matplotlib.pylab as pylab
from skimage.segmentation import (morphological_chan_vese,morphological_geodesic_active_contour,inverse_gaussian_gradient, checkerboard_level_set)
img=cv2.imread('C:/Users/xpp/Desktop/Lena.png')#原始图像
newmask=cv2.imread('C:/Users/xpp/Desktop/mask.png')#原始图像
mask=2*np.ones(img.shape[:2],np.uint8)
mask[(newmask[...,0]<=76)&(newmask[...,1]<=76)&(newmask[...,2]>=76)]=0
mask[(newmask[...,0]<=76)&(newmask[...,1]>=76)&(newmask[...,2]<=76)]=1
mask,bg_model,fg_model=cv2.grabCut(img,mask,None,bg_model,fg_model,5,cv2.GC_INIT_WITH_MASK)#grabcut图像分割
mask=np.where((mask==2)|(mask==0),0,1).astype('uint8')
img=img*mask[:,:,np.newaxis]
pylab.imshow(img)
pylab.colorbar()
pylab.show()