GrabCut
该算法,在opencv中很容易就可以使用。
通过对于一开始取定的矩阵来做来模拟。
(这个我们可以通过调整这个点的以及对应的矩阵大小来调整效果)
下面就是算法提取出来的图像,以及原图
下面这就是原图(就是代码的输入图片)
代码
import cv2
import numpy as np
from matplotlib import pyplot as plt
img = cv2.imread('3.jpg')
OLD_IMG = img.copy()
mask = np.zeros(img.shape[:2], np.uint8