OpenCV中用GrabCut进行前景检测

最新推荐文章于 2024-08-16 20:20:08 发布

计算机视觉与OpenCV

最新推荐文章于 2024-08-16 20:20:08 发布

阅读量661

点赞数 1

分类专栏：学习OpenCV 4：基于Python的算法实战文章标签： opencv 前景检测深度学习计算机视觉

本文链接：https://blog.csdn.net/lxiao428/article/details/86515088

版权

学习OpenCV 4：基于Python的算法实战专栏收录该内容

51 篇文章 83 订阅

订阅专栏

import cv2
import numpy as np
import matplotlib.pyplot as plt
img = cv2.imread("img3.jpg")
mask = np.zeros(img.shape[:2], np.uint8) #与图片同大小的掩模
bgdModel = np.zeros((1,65), np.float64)
fgdModel = np.zeros((1,65), np.float64)

rect = (30, 64, 536, 890) #定义矩形区域，区域外的被认为是背景，区域中的部分会识别是背景还是前景
cv2.grabCut(img, mask,rect, bgdModel, fgdModel, 50, 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.show()

检测结果：