Dalam Python, imej boleh dibahagikan dan diekstrak menggunakan kaedah daripada perpustakaan OpenCV.

Pembahagian dan pengekstrakan imej

Segmen atau ekstrak objek latar depan dalam imej sebagai imej sasaran. Latar belakang itu sendiri tidak berminat Algoritma aliran air dan segmen algoritma GrabCut dan mengekstrak imej.

Gunakan algoritma aliran air untuk mencapai pembahagian dan pengekstrakan imej

Algoritma aliran air membandingkan imej dengan permukaan rupa bumi geografi untuk mencapai pembahagian imej ini sangat berkesan.

Prinsip Algoritma

Sebarang imej skala kelabu boleh dianggap sebagai permukaan rupa bumi geografi Kawasan dengan nilai skala kelabu yang tinggi boleh dianggap sebagai nilai skala kelabu Kawasan yang lebih rendah boleh dianggap sebagai lembah.

Imej kiri ialah imej asal dan imej kanan ialah "permukaan topografi" yang sepadan dengannya.

Proses ini memisahkan imej kepada dua set berbeza: lembangan tadahan dan garisan tadahan air. Empangan yang kami bina ialah garisan aliran air, iaitu pembahagian imej asal. Ini adalah algoritma tadahan air.

Disebabkan oleh pengaruh hingar dan faktor lain, algoritma asas tadahan air di atas sering mengakibatkan pembahagian berlebihan. Pembahagian berlebihan akan membahagikan imej kepada blok kecil bebas yang padat, menjadikan pembahagian tidak bermakna. Untuk menambah baik kesan pembahagian imej, algoritma tadahan air berasaskan topeng telah dicadangkan. Algoritma tadahan air yang dipertingkatkan membolehkan pengguna melabelkan bahagian yang difikirkannya adalah kawasan bersegmen yang sama (bahagian yang dilabel dipanggil topeng). Apabila algoritma tadahan air diproses, bahagian yang ditanda akan diproses ke dalam kawasan bersegmen yang sama.

Contohnya:

Imej asal diberi anotasi dan ditandakan sebagai tiga blok warna kecil yang ditandakan sebagai warna gelap Apabila menggunakan algoritma tadahan air topeng , warna yang terkandung di dalamnya bahagian akan dibahagikan kepada kawasan yang sama.

Pengenalan kepada fungsi berkaitan

Dalam OpenCV, anda boleh menggunakan fungsi cv2.watershed() untuk melaksanakan algoritma watershed.

Dalam proses pelaksanaan khusus, ia juga perlu menggunakan fungsi morfologi, fungsi transformasi jarak cv2.distanceTransform(), cv2.connectedComponents() untuk melengkapkan pembahagian imej.

• Fungsi morfologi
  Sebelum menggunakan algoritma aliran air untuk membahagikan imej, pemprosesan morfologi imej yang ringkas diperlukan.

• Operasi pembukaan
  Operasi pembukaan ialah operasi yang menghakis dahulu dan kemudian mengembang boleh mengeluarkan bunyi dalam imej
  Ia diproses oleh algoritma aliran air Sebelum mengambil imej, kita mesti terlebih dahulu menggunakan operasi pembukaan untuk mengeluarkan bunyi dalam imej untuk mengelakkan gangguan yang mungkin disebabkan oleh hingar pada pembahagian imej.

• Dapatkan sempadan imej
  Sempadan imej boleh diperoleh melalui operasi morfologi dan operasi tolak.
  Gunakan transformasi morfologi untuk mendapatkan maklumat sempadan imej

import cv2
import numpy as np
import matplotlib.pyplot as plt
o=cv2.imread("my.bmp", cv2.IMREAD_UNCHANGED)
k=np.ones((5,5), np.uint8)
e=cv2.erode(o, k)
b=cv2.subtract(o, e)
plt.subplot(131)
plt.imshow(o)
plt.axis('off')
plt.subplot(132)
plt.imshow(e)
plt.axis('off')
plt.subplot(133)
plt.imshow(b)
plt.axis('off')
plt.show()

Gunakan operasi morfologi dan operasi tolak untuk mendapatkan maklumat sempadan imej. Walau bagaimanapun, operasi morfologi hanya sesuai untuk imej yang agak mudah. Jika objek latar depan dalam imej disambungkan, sempadan setiap sub-imej tidak boleh diperoleh dengan tepat menggunakan operasi morfologi.

• Fungsi Transform Jarak distanceTransform
  Apabila sub-imej dalam imej tidak disambungkan, operasi hakisan morfologi boleh digunakan terus untuk menentukan objek latar depan, tetapi jika sub- imej dalam imej disambungkan Apabila ia bersama, sukar untuk menentukan objek latar depan
  Pada masa ini, objek latar depan boleh diekstrak dengan mudah dengan bantuan fungsi transformasi jarak cv2.distanceTransform().
  Fungsi cv2.distanceTransform() mengira jarak dari mana-mana titik dalam imej binari ke titik latar belakang terdekat.
  Secara amnya, fungsi ini mengira jarak dari piksel bukan nilai sifar dalam imej ke piksel nilai sifar terdekat, iaitu, ia mengira jarak antara semua piksel dalam imej binari dan piksel terdekatnya dengan nilai 0. .

Jika nilai piksel itu sendiri ialah 0, maka jaraknya juga 0.

• Hasil pengiraan cv2.distanceTransform() menggambarkan hubungan jarak antara setiap piksel dan latar belakang (titik piksel dengan nilai 0).

Biasanya:

• Jika pusat (pusat jisim) objek latar depan adalah 0 piksel jaraknya semakin jauh anda pergi, semakin besar nilainya.

• Jika tepi objek latar depan lebih dekat dengan piksel dengan nilai 0, nilai yang lebih kecil akan diperolehi.

Jika keputusan pengiraan di atas adalah ambang, pusat, rangka dan maklumat lain sub-imej dalam imej boleh diperolehi. Fungsi transformasi jarak cv2.distanceTransform() boleh digunakan untuk mengira pusat objek, dan juga boleh memperhalusi garis besar, mendapatkan latar depan imej, dsb.

Format sintaks fungsi cv2 .distanceTransform() ialah:

dst=cv2.distanceTransform(src, distanceType, maskSize[, dstType]])

• src ialah imej binari saluran tunggal 8-bit.

• distanceType ialah parameter jenis jarak

• maskSize为掩模的尺寸
  distanceType=cv2.DIST_L1或cv2.DIST_C时，maskSize强制为3（因为设置为3和设置为5及更大值没有什么区别）。

• dstType为目标图像的类型，默认值为CV_32F。

• dst表示计算得到的目标图像，可以是8位或32位浮点数，尺寸和src相同。

使用距离变换函数cv2.distanceTransform()，计算一幅图像的确定前景

import numpy as np
import cv2
import matplotlib.pyplot as plt
img = cv2.imread('water_coins.jpg')
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

img=cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
ishow=img.copy()
ret, thresh = cv2.threshold(gray,0,255,cv2.THRESH_BINARY_INV+cv2.THRESH_OTSU)
kernel = np.ones((3,3), np.uint8)
opening = cv2.morphologyEx(thresh, cv2.MORPH_OPEN, kernel, iterations = 2) # 进行开运算
dist_transform = cv2.distanceTransform(opening, cv2.DIST_L2,5)
ret, fore = cv2.threshold(dist_transform,0.7*dist_transform.max(),255,0)
plt.subplot(131)
plt.imshow(ishow)
plt.axis('off')
plt.subplot(132)
plt.imshow(dist_transform)
plt.axis('off')
plt.subplot(133)
plt.imshow(fore)
plt.axis('off')
plt.show()

fore图像中： 比较准确地显示出左图内的“确定前景”。确定前景，通常是指前景对象的中心。之所以认为这些点是确定前景，是因为它们距离背景点的距离足够远，都是距离大于足够大的固定阈值（0.7*dist_transform.max()）的点。

• 确定未知区域
  使用形态学的膨胀操作能够将图像内的前景“膨胀放大”。
  当图像内的前景被放大后，背景就会被“压缩”，所以此时得到的背景信息一定小于实际背景的，不包含前景的“确定背景”。

为了方便说明将确定背景称为B。

• 距离变换函数cv2.distanceTransform()能够获取图像的“中心”，得到“确定前景”。
  图像中有了确定前景F和确定背景B，剩下区域的就是未知区域UN了。这部分区域正是分水岭算法要进一步明确的区域。
  针对一幅图像O，通过以下关系能够得到未知区域UN：

• 未知区域UN=图像O-确定背景B-确定前景F

• 未知区域UN=（图像O-确定背景B）- 确定前景F

“图像O-确定背景B”，可以通过对图像进行形态学的膨胀操作得到。

标注一幅图像的确定前景、确定背景及未知区域。

import numpy as np
import cv2
import matplotlib.pyplot as plt
img = cv2.imread('water_coins.jpg')
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
img=cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
ishow=img.copy()
ret, thresh = cv2.threshold(gray,0,255,cv2.THRESH_BINARY_INV+cv2.THRESH_OTSU)
kernel = np.ones((3,3), np.uint8)
opening = cv2.morphologyEx(thresh, cv2.MORPH_OPEN, kernel, iterations = 2)
bg = cv2.dilate(opening, kernel, iterations=3)
dist = cv2.distanceTransform(opening, cv2.DIST_L2,5)
ret, fore = cv2.threshold(dist,0.7*dist.max(),255,0)
fore = np.uint8(fore)
un = cv2.subtract(bg, fore)
plt.subplot(221)
plt.imshow(ishow)
plt.axis('off')
plt.subplot(222)
plt.imshow(bg)
plt.axis('off')
plt.subplot(223)
plt.imshow(fore)
plt.axis('off')
plt.subplot(224)
plt.imshow(un)
plt.axis('off')
plt.show()

• 函数connectedComponents
  明确了确定前景后，就可以对确定前景图像进行标注了。
  在OpenCV中，可以使用函数cv2.connectedComponents()进行标注。该函数会将背景标注为0，将其他的对象使用从1开始的正整数标注。
  函数cv2.connectedComponents()的语法格式为：

retval, labels = cv2.connectedComponents( image )

• image为8位单通道的待标注图像。

• retval为返回的标注的数量。

• labels为标注的结果图像。

使用函数cv2.connectedComponents()标注一幅图像

import numpy as np
import cv2
import matplotlib.pyplot as plt
img = cv2.imread('water_coins.jpg')
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
img=cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
ishow=img.copy()
ret, thresh = cv2.threshold(gray,0,255,cv2.THRESH_BINARY_INV+cv2.THRESH_OTSU)
kernel = np.ones((3,3), np.uint8)
opening = cv2.morphologyEx(thresh, cv2.MORPH_OPEN, kernel, iterations = 2)
sure_bg = cv2.dilate(opening, kernel, iterations=3)
dist_transform = cv2.distanceTransform(opening, cv2.DIST_L2,5)
ret, fore = cv2.threshold(dist_transform,0.7*dist_transform.max(),255,0)
fore = np.uint8(fore)
ret, markers = cv2.connectedComponents(fore)
print(ret)
plt.subplot(131)
plt.imshow(ishow)
plt.axis('off')
plt.subplot(132)
plt.imshow(fore)
plt.axis('off')
plt.subplot(133)
plt.imshow(markers)
plt.axis('off')
plt.show()

前景图像的中心点被做了不同的标注（用不同颜色区分)

函数cv2.connectedComponents()在标注图像时，会将背景标注为0，将其他的对象用从1开始的正整数标注。具体的对应关系为：

• 数值0代表背景区域。

• 从数值1开始的值，代表不同的前景区域。

在分水岭算法中，标注值0代表未知区域。所以，我们要对函数cv2.connectedComponents()标注的结果进行调整：将标注的结果都加上数值1。经过上述处理后，在标注结果中：

• 数值1代表背景区域。

• 从数值2开始的值，代表不同的前景区域。

为了能够使用分水岭算法，还需要对原始图像内的未知区域进行标注，将已经计算出来的未知区域标注为0即可。

关键代码：

ret, markers = cv2.connectedComponents(fore)
markers = markers+1
markers[未知区域] = 0

使用函数cv2.connectedComponents()标注一幅图像，并对其进行修正，使未知区域被标注为0

import numpy as np
import cv2
import matplotlib.pyplot as plt
img = cv2.imread('water_coins.jpg')
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
img=cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
ishow=img.copy()
ret, thresh = cv2.threshold(gray,0,255,cv2.THRESH_BINARY_INV+cv2.THRESH_OTSU)
kernel = np.ones((3,3), np.uint8)
opening = cv2.morphologyEx(thresh, cv2.MORPH_OPEN, kernel, iterations = 2)
sure_bg = cv2.dilate(opening, kernel, iterations=3)
dist_transform = cv2.distanceTransform(opening, cv2.DIST_L2,5)
ret, fore = cv2.threshold(dist_transform,0.7*dist_transform.max(),255,0)
fore = np.uint8(fore)
ret, markers1 = cv2.connectedComponents(fore)
foreAdv=fore.copy()
unknown = cv2.subtract(sure_bg, foreAdv)
ret, markers2 = cv2.connectedComponents(foreAdv)
markers2 = markers2+1
markers2[unknown==255] = 0
plt.subplot(121)
plt.imshow(markers1)
plt.axis('off')
plt.subplot(122)
plt.imshow(markers2)
plt.axis('off')
plt.show()

前景都有一个黑色的边缘，这个边缘是被标注的未知区域。

• 函数cv2.watershed()
  完成上述处理后，就可以使用分水岭算法对预处理结果图像进行分割了。
  在OpenCV中，实现分水岭算法的函数是cv2.watershed()，其语法格式为：

markers = cv2.watershed( image, markers )

• image是输入图像，必须是8位三通道的图像。在对图像使用

cv2.watershed()函数处理之前，必须先用正数大致勾画出图像中的期望分割区域。每一个分割的区域会被标注为1、2、3等。对于尚未确定的区域，需要将它们标注为0。我们可以将标注区域理解为进行分水岭算法分割的“种子”区域。

• markers是32位单通道的标注结果，它应该和image具有相等大小。在markers中，每一个像素要么被设置为初期的“种子值”，要么被设置为**“-1”表示边界**。

分水岭算法图像分割实例

使用分水岭算法进行图像分割时，基本的步骤为：

• 通过形态学开运算对原始图像O去噪。

• 通过腐蚀操作获取“确定背景B”。
  需要注意，这里得到“原始图像-确定背景”即可。

• 利用距离变换函数cv2.distanceTransform()对原始图像进行运算，并对其进行阈值处理，得到“确定前景F”。

• 计算未知区域UN（UN=O -B - F）

• 利用函数cv2.connectedComponents()对原始图像O进行标注。

• 对函数cv2.connectedComponents()的标注结果进行修正。

• 使用分水岭函数完成对图像的分割。

使用分水岭算法对一幅图像进行分割：

import numpy as np
import cv2
import matplotlib.pyplot as plt
img = cv2.imread('water_coins.jpg')
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
img=cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
ishow=img.copy()
ret, thresh = cv2.threshold(gray,0,255,cv2.THRESH_BINARY_INV+cv2.THRESH_OTSU)
kernel = np.ones((3,3), np.uint8)
opening = cv2.morphologyEx(thresh, cv2.MORPH_OPEN, kernel, iterations = 2)
sure_bg = cv2.dilate(opening, kernel, iterations=3)
dist_transform = cv2.distanceTransform(opening, cv2.DIST_L2,5)
ret, sure_fg = cv2.threshold(dist_transform,0.7*dist_transform.max(),255,0)
sure_fg = np.uint8(sure_fg)
unknown = cv2.subtract(sure_bg, sure_fg)
ret, markers = cv2.connectedComponents(sure_fg)
markers = markers+1
markers[unknown==255] = 0
markers = cv2.watershed(img, markers)
img[markers == -1] = [0,255,0] # 边界
plt.subplot(121)
plt.imshow(ishow)
plt.axis('off')
plt.subplot(122)
plt.imshow(img)
plt.axis('off')
plt.show()

交互式前景提取

经典的前景提取技术主要使用纹理（颜色）信息，如魔术棒工具，或根据边缘（对比度）信息，如智能剪刀等。在开始提取前景时，先用一个矩形框指定前景区域所在的大致位置范围，然后不断迭代地分割，直到达到最好的效果。经过上述处理后，提取前景的效果可能并不理想，存在前景没有提取出来，或者将背景提取为前景的情况，此时需要用户干预提取过程。

用户在原始图像的副本中（也可以是与原始图像大小相等的任意一幅图像），用白色标注要提取为前景的区域，用黑色标注要作为背景的区域。然后，将标注后的图像作为掩模，让算法继续迭代提取前景从而得到最终结果。

GrabCut算法的具体实施过程。

• 将前景所在的大致位置使用矩形框标注出来。

• 此时矩形框框出的仅仅是前景的大致位置，其中既包含前景又包含背景，所以该区域实际上是未确定区域。但是，该区域以外的区域被认为是“确定背景”。

• 根据矩形框外部的“确定背景”数据来区分矩形框区域内的前景和背景。

• 用高斯混合模型（Gaussians Mixture Model, GMM）对前景和背景建模。

• GMM会根据用户的输入 学习并创建新的像素分布。对未分类的像素（可能是背景也可能是前景），根据其与已知分类像素（前景和背景）的关系进行分类。

• 根据像素分布情况生成一幅图，图中的节点就是各个像素点。

• 除了像素点之外，还有两个节点：前景节点和背景节点。所有的前景像素都和前景节点相连，所有的背景像素都和背景节点相连。每个像素连接到前景节点或背景节点的边的权重由像素是前景或背景的概率来决定。

• 图中的每个像素除了与前景节点或背景节点相连外，彼此之间还存在着连接。两个像素连接的边的权重值由它们的相似性决定，两个像素的颜色越接近，边的权重值越大。

• 完成节点连接后，需要解决的问题变成了一幅连通的图。在该图上根据各自边的权重关系进行切割，将不同的点划分为前景节点和背景节点。

• 不断重复上述过程，直至分类收敛为止。

在OpenCV中，实现交互式前景提取的函数是cv2.grabCut()，其语法格式为：

mask, bgdModel, fgdModel =cv2.grabCut(img, mask, rect, bgdModel, fgdModel, iterCount[, mode] )

• img为输入图像，要求是8位3通道的。

• mask为掩模图像，要求是8位单通道的。该参数用于确定前景区域、背景区域和不确定区域，可以设置为4种形式。

• cv2.GC_BGD：表示确定背景，也可以用数值0表示。

• cv2.GC_FGD：表示确定前景，也可以用数值1表示。

• cv2.GC_PR_BGD：表示可能的背景，也可以用数值2表示。

• cv2.GC_PR_FGD：表示可能的前景，也可以用数值3表示。

在最后使用模板提取前景时，会将参数值0和2合并为背景（均当作0处理），将参数值1和3合并为前景（均当作1处理）。

在通常情况下，我们可以使用白色笔刷和黑色笔刷在掩模图像上做标记，再通过转换将其中的白色像素设置为0，黑色像素设置为1。

• rect指包含前景对象的区域，该区域外的部分被认为是“确定背景”。因此，在选取时务必确保让前景包含在rect指定的范围内；否则，rect外的前景部分是不会被提取出来的。

只有当参数mode的值被设置为矩形模式cv2.GC_INIT_WITH_RECT时，参数rect才有意义。

其格式为（x, y, w, h），分别表示区域左上角像素的x轴和y轴坐标以及区域的宽度和高度。

如果前景位于右下方，又不想判断原始图像的大小，对于w 和h可以直接用一个很大的值。

使用掩模模式时，将该值设置为none即可。

• bgdModel为算法内部使用的数组，只需要创建大小为(1, 65)的numpy.float64数组。

• fgdModel为算法内部使用的数组，只需要创建大小为(1, 65)的numpy.float64数组。

• iterCount表示迭代的次数。

mode表示迭代模式。其可能的值与含义如下：

RECT 和MASK可以组合使用( 并的关系 )

使用GrabCut算法提取图像的前景

import numpy as np
import cv2
import matplotlib.pyplot as plt
o = cv2.imread('lenacolor.png')
orgb=cv2.cvtColor(o, cv2.COLOR_BGR2RGB)
mask = np.zeros(o.shape[:2], np.uint8)
bgdModel = np.zeros((1,65), np.float64)
fgdModel = np.zeros((1,65), np.float64)
rect = (50,50,400,500)
cv2.grabCut(o, mask, rect, bgdModel, fgdModel,5, cv2.GC_INIT_WITH_RECT)
mask2 = np.where((mask==2)|(mask==0),0,1).astype('uint8')
ogc = o*mask2[:, :, np.newaxis]
ogc=cv2.cvtColor(ogc, cv2.COLOR_BGR2RGB)
plt.subplot(121)
plt.imshow(orgb)
plt.axis('off')
plt.subplot(122)
plt.imshow(ogc)
plt.axis('off')
plt.show()

为了得到完整的前景对象，需要做一些改进。

这里对原始图像进行标注，将需要保留的部分设置为白色，将需要删除的背景设置为黑色。以标记好的图像作为模板，使用函数cv2.grabCut()完成前景的提取。

这个过程主要包含以下步骤:

• 利用函数cv2.grabCut()在cv2.GC_INIT_WITH_RECT 模式下对图像进行初步的前景提取，得到初步提取的结果图像og。

• 使用Windows系统自带的笔刷工具，打开要提取前景的图像，比如lena。

• 使用白色笔刷在希望提取的前景区域做标记。

• 使用黑色笔刷在希望删除的背景区域做标记。

• 将当前设置好的lena图像另存为模板图像m0。

• 将模板图像m0中的白色值和黑色值映射到模板m中。将模板图像m0中的白色值（像素值为255）映射为模板图像m中的确定前景（像素值为1），将模板图像m0中的黑色值（像素值为0）映射为模板图像m中的确定背景（像素值为0）。

• 以模板图像m作为函数cv2.grabCut()的模板参数（mask），对图像og完成前景提取。
  使用画笔标记的模板图像m0不能直接作为模板（即参数mask）使用
  
  函数cv2.grabCut()要求，参数mask的值必须是cv2.GC_BGD（确定背景）、cv2.GC_FGD（确定前景）、cv2.GC_PR_BGD（可能的背景）、cv2.GC_PR_FGD（可能的前景），或者是0、1、2、3之中的值。
  必须先将模板图像m0中的白色值和黑色值映射到模板m上，再将模板图像m作为函数cv2.grabCut()的模板参数。

在GrabCut算法中使用模板提取图像的前景:

import numpy as np
import cv2
import matplotlib.pyplot as plt
o= cv2.imread('lenacolor.png')
orgb=cv2.cvtColor(o, cv2.COLOR_BGR2RGB)
mask = np.zeros(o.shape[:2], np.uint8)
bgd = np.zeros((1,65), np.float64)
fgd = np.zeros((1,65), np.float64)
rect = (50,50,400,500)
cv2.grabCut(o, mask, rect, bgd, fgd,5, cv2.GC_INIT_WITH_RECT)
mask2 = cv2.imread('mask.png',0)
mask2Show = cv2.imread('mask.png', -1)
m2rgb=cv2.cvtColor(mask2Show, cv2.COLOR_BGR2RGB)
mask[mask2 == 0] = 0
mask[mask2 == 255] = 1
mask, bgd, fgd = cv2.grabCut(o, mask, None, bgd, fgd,5, cv2.GC_INIT_WITH_MASK)
mask = np.where((mask==2)|(mask==0),0,1).astype('uint8')
ogc = o*mask[:, :, np.newaxis]
ogc=cv2.cvtColor(ogc, cv2.COLOR_BGR2RGB)
plt.subplot(121)
plt.imshow(m2rgb)
plt.axis('off')
plt.subplot(122)
plt.imshow(ogc)
plt.axis('off')
plt.show()

在函数cv2.grabCut()的实际使用中，也可以不使用矩形初始化，直接使用模板模式。构造一个模板图像，其中：

• 使用像素值0标注确定背景。

• 使用像素值1标注确定前景。

• 使用像素值2标注可能的背景。

• 使用像素值3标注可能的前景。

构造完模板后，直接将该模板用于函数cv2.grabCut()处理原始图像，即可完成前景的提取。

一般情况下，自定义模板的步骤为：

• 先使用numpy.zeros构造一个内部像素值都是0（表示确定背景）的图像mask，以便在后续步骤中逐步对该模板图像进行细化。

• ．使用mask[30:512, 50:400]=3，将模板图像中第30行到第512行，第50列到400列的区域划分为可能的前景（像素值为3，对应参数mask的含义为“可能的前景”）。

• 使用mask[50:300, 150:200]=1，将模板图像中第50行到第300行，第150列到第200列的区域划分为确定前景（像素值为1，对应参数mask的含义为“确定前景”）。

在GrabCut算法中直接使用自定义模板提取图像的前景

import numpy as np
import cv2
import matplotlib.pyplot as plt
o= cv2.imread('lenacolor.png')
orgb=cv2.cvtColor(o, cv2.COLOR_BGR2RGB)
bgd = np.zeros((1,65), np.float64)
fgd = np.zeros((1,65), np.float64)
mask2 = np.zeros(o.shape[:2], np.uint8)
#先将掩模的值全部构造为0（确定背景），在后续步骤中，再根据需要修改其中的部分值
mask2[30:512,50:400]=3 #lena头像的可能区域
mask2[50:300,150:200]=1 #lena头像的确定区域，如果不设置这个区域，头像的提取不完整
cv2.grabCut(o, mask2, None, bgd, fgd,5, cv2.GC_INIT_WITH_MASK)
mask2 = np.where((mask2==2)|(mask2==0),0,1).astype('uint8')
ogc = o*mask2[:, :, np.newaxis]
ogc=cv2.cvtColor(ogc, cv2.COLOR_BGR2RGB)
plt.subplot(121)
plt.imshow(orgb)
plt.axis('off')
plt.subplot(122)
plt.imshow(ogc)
plt.axis('off')
plt.show()

对于不同的图像，要构造不同的模板来划分它们的确定前景、确定背景、可能的前景与可能的背景。

Atas ialah kandungan terperinci Dalam Python, imej boleh dibahagikan dan diekstrak menggunakan kaedah daripada perpustakaan OpenCV.. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!