You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
31 lines
937 B
31 lines
937 B
6 months ago
|
import cv2
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
src = cv2.imread("demo2.jpg")
|
||
|
|
r = cv2.selectROI('input', src, False) # 返回 (x_min, y_min, w, h)
|
||
|
|
# roi区域
|
||
|
|
roi = src[int(r[1]):int(r[1]+r[3]), int(r[0]):int(r[0]+r[2])]
|
||
|
|
# 原图mask
|
||
|
|
mask = np.zeros(src.shape[:2], dtype=np.uint8)
|
||
|
|
# 矩形roi
|
||
|
|
rect = (int(r[0]), int(r[1]), int(r[2]), int(r[3])) # 包括前景的矩形，格式为(x,y,w,h)
|
||
|
|
|
||
|
|
bgdmodel = np.zeros((1,65),np.float64) # bg模型的临时数组
|
||
|
|
fgdmodel = np.zeros((1,65),np.float64) # fg模型的临时数组
|
||
|
|
|
||
|
|
cv2.grabCut(src,mask,rect,bgdmodel,fgdmodel, 11, mode=cv2.GC_INIT_WITH_RECT)
|
||
|
|
|
||
|
|
# 提取前景和可能的前景区域
|
||
|
|
mask2 = np.where((mask==1) + (mask==3), 255, 0).astype('uint8')
|
||
|
|
|
||
|
|
print(mask2.shape)
|
||
|
|
|
||
|
|
result = cv2.bitwise_and(src,src,mask=mask2)
|
||
|
|
cv2.imwrite('result.jpg', result)
|
||
|
|
cv2.imwrite('roi.jpg', roi)
|
||
|
|
|
||
|
|
cv2.imshow('roi', roi)
|
||
|
|
cv2.imshow("result", result)
|
||
|
|
cv2.waitKey(0)
|
||
|
|
cv2.destroyAllWindows()
|