# -*- coding: utf-8 -*-

import cv2
import scipy
import scipy.ndimage
from PIL import Image

img = scipy.array(Image.open('image.png'))
mask = scipy.array(Image.open('mask.png'))
assert img.shape == mask.shape
imgmask = scipy.zeros((mask.shape[0],mask.shape[1]),scipy.uint8)
imgmask[mask[...,0] == 0] = cv2.GC_BGD
imgmask[mask[...,0] == 64] = cv2.GC_PR_BGD
imgmask[mask[...,0] == 128] = cv2.GC_PR_FGD
imgmask[mask[...,0] == 255] = cv2.GC_FGD
print mask.shape
print imgmask.shape
bgdModel = scipy.zeros((1,5*13),scipy.float64)
fgdModel = scipy.zeros((1,5*13),scipy.float64)
cv2.grabCut(
    img,                    
    imgmask,                
    None,                   
    bgdModel,               
    fgdModel,               
    1,  
    cv2.GC_INIT_WITH_MASK,
    )
mask = (imgmask == cv2.GC_FGD) | (imgmask == cv2.GC_PR_FGD)
masked = scipy.zeros_like(img)
masked[mask] = img[mask]
cv2.imshow('test',masked)
cv2.waitKey(0)