吾爱破jie 发表于 2019-7-6 22:43

Grabcut图像分割

本帖最后由 吾爱破jie 于 2019-7-6 22:44 编辑

import numpy as np
import cv2
from matplotlib importpyplot as plt
img = cv2.imread('4.jpg')
mask = np.zeros(img.shape[:2],np.uint8)

bgdModel = np.zeros((1,65),np.float64)
fgdModel = np.zeros((1,65),np.float64)

rect = (175,250,421,378)
cv2.grabCut(img,mask,rect,bgdModel,fgdModel,5,cv2.GC_INIT_WITH_RECT)

mask2 = np.where((mask==2)|(mask==0),0,1).astype('uint8')
img = img*mask2[:,:,np.newaxis]

plt.subplot(121),plt.imshow(img)
plt.title('grabcut'),plt.xticks([]),plt.yticks([])
plt.subplot(122),plt.imshow(cv2.cvtColor(cv2.imread('4.jpg'),cv2.COLOR_BGR2RGB))
plt.title('original'),plt.xticks([]),plt.yticks([])
plt.show()






在这里我想知道我该如何编辑矩形框的大小,从而让这个建筑物被完整的分割出来。
rect = (175,250,421,378)这个应该是控制矩形框大小的,他的四个参数分别代表什么,我该如何调整他呢,请了解的大佬帮我解释一下。

wtujoxk 发表于 2019-7-6 23:13

https://docs.opencv.org/3.4.2/d8/d83/tutorial_py_grabcut.html

qianyu666 发表于 2019-7-7 01:10

不错,学习了

吾爱破jie 发表于 2019-7-7 08:48

qianyu666 发表于 2019-7-7 01:10
不错,学习了

。。。。。。。
页: [1]
查看完整版本: Grabcut图像分割