GrabCut Foreground Extraction OpenCV Python Tutorial

Selamat Datang di tutorial ekstraksi foreground dengan OpenCV dan Python . Idenya di sini adalah untuk menemukan latar depan , dan menghapus latar belakang .
Untuk memulai , kita akan menggunakan gambar :



Mari kita Load gambar dan mendefinisi beberapa hal :

import numpy as np
import cv2
from matplotlib import pyplot as plt

img = cv2.imread('opencv-python-foreground-extraction-tutorial.jpg')
mask = np.zeros(img.shape[:2],np.uint8)

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

rect = (161,79,150,150)

kami telah impor cv2 , numpy , dan matplotlib . Kemudian kami memuat dalam gambar , membuat mask , tentukan latar belakang dan model foreground , yang digunakan oleh algoritma internal . Bagian penting sebenarnya adalah mendefinisakn rect kita . Ini adalaha rect = ( start_x , start_y , lebar, tinggi ) .

Ini adalah persegi panjang yang mengenai objek utama kami . Jika Anda menggunakan gambar saya , itu adalah rect yangdigunakan. Jika Anda menggunakan Anda sendiri , temukan koordinat yang tepat untuk gambar Anda .

Berikutnya:

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.imshow(img)
plt.colorbar()
plt.show()

Jadi di sini kami menggunakan cv2.grabCut , yang mengambil beberapa parameter . Pertama input gambar , kemudian mask , kemudian persegi panjang untuk objek utama kami , model latar belakang , Model foreground , jumlah iterasi untuk menjalankan , dan mode apa yang Anda gunakan .

Dari sini , mask berubah sehingga semua 0 dan 2 piksel dikonversi ke latar belakang , di mana 1 dan 3 piksel sekarang latar depan . Dari sini , kita kalikan dengan gambar input , dan kami mendapatkan hasil akhir kami :

Tampilan Hasil :