1
我有一个图像,橙色和白色背景。我想让白色背景变得透明。下面的代码使用grabcut来制作一个面具。然后,我将图像分割成RGB通道并在Alpha通道上应用蒙版。您会从下面的图片中看到,后期截取和遮罩图片都可以。我一直无法弄清楚如何将遮罩应用于Alpha通道。建议感激。使图像背景透明
im = cv2.imread(sourceimagefile)
cv2.imshow('original',im)
mask = np.zeros(im.shape[:2],np.uint8)
rect = (box[0][0], box[0][1], box[0][2]-box[0][0], box[0][3]-box[0][1])
bgdModel = np.zeros((1,65),np.float64)
fgdModel = np.zeros((1,65),np.float64)
cv2.grabCut(im,mask,rect,bgdModel,fgdModel,5,cv2.GC_INIT_WITH_RECT)
if len(np.where((mask==3)|(mask==1))[0])>0:
mask2 = np.where((mask==2)|(mask==0),0,1).astype('uint8')
mask2 = np.repeat(mask2[:,:,np.newaxis],3,axis=2)
else:
mask2 = np.zeros_like(im)
mask2[box[0][1]:box[0][3],box[0][0]:box[0][2],:] = 1
im2 = im*mask2
cv2.imshow('post-grabcut',im2)
minVal, maxVal, minLoc, maxLoc = cv2.minMaxLoc(mask)
flag, mask = cv2.threshold(mask, maxVal-1, 255, cv2.cv.CV_THRESH_BINARY)
cv2.imshow("mask", mask)
b, g, r = cv2.split(im2)
img_RGBA = cv2.merge((b, g, r, mask))
cv2.imshow("final",img_RGBA)
我原来的代码是正确的,掩码* 255不是必需的。问题出在您正确指出的imshow不支持alpha通道。你想编辑你的答案,我会接受吗? – gobob
我做了更改,很高兴看到一些信息对您有帮助@gobob – andrew