前面学习过分水岭算法来分离背景与前景,现在来学习另一个标记算法GrabCut,它也是通过标记来抽取前景的算法。这个算法与分水岭算法标记不一样,比如只要简单地标记一个前景的方框,就可以抽取出来,不过计算量就有点大了。如下图:
在这里使用红色的矩形把中间的烧鸭选出来,就这样就可以把它放到OpenCV里的函数grabCut去处理,就会把烧鸭分离出来,既然分离出来了,你就可以进行各种处理了。整个代码如下:
#python 3.7.4,opencv4.1
#蔡军生 https://blog.csdn.net/caimouse/article/details/51749579
#
import numpy as np
import cv2
from matplotlib import pyplot as plt
img = cv2.imread('yaze1.png')
mask = np.zeros(img.shape[:2],np.uint8)