直接上代码,原理大家都可以查到
注意:以下img自己加载即可,我是事先把图片转换成了单通道的Array形式,并且我的img是归一化后在0-1的。生成的label只有0,1两个值。
import cv2 as cv
import numpy as np
img = np.uint8(img*255)
src = img
r = cv.selectROI('input', img, False)
roi = src[int(r[1]):int(r[1] + r[3]), int(r[0]):int(r[0] + r[2])]
maskc = np.zeros(src.shape[:2], dtype=np.uint8)
rect = (int(r[0]), int(r[1]), int(r[2]), int(r[3]))
bgdmodel = np.zeros((1, 65), np.float64)
fgdmodel = np.zeros((1, 65), np.float64)
imgc = cv.cvtColor(src, cv.COLOR_GRAY2BGR)
cv.grabCut(imgc, maskc, rect, bgdmodel, fgdmodel, 11, mode=cv.GC_INIT_WITH_RECT)
mask2 = np.where((maskc == 1) + (maskc == 3), 255, 0).astype('uint8')
print(mask2.shape)
result = cv.bitwise_and(src, src, mask=mask2)
cv.destroyAllWindows()
label = np.uint8(result>0)