Skip to content

Instantly share code, notes, and snippets.

@Lucs1590
Created September 4, 2019 20:29
Show Gist options
  • Save Lucs1590/88098c6bf70f1d0a1532958dcd33105f to your computer and use it in GitHub Desktop.
Save Lucs1590/88098c6bf70f1d0a1532958dcd33105f to your computer and use it in GitHub Desktop.
To create a black background in an image
import cv2
import numpy as np
import utils as ut
from matplotlib import pyplot as plt
img = '/home/brito/Documentos/Dev/tcc/img/f1.jpeg'
img = cv2.imread(img)
mask = np.zeros(img.shape[:2],np.uint8)
bgdModel = np.zeros((1,65),np.float64)
fgdModel = np.zeros((1,65),np.float64)
rect = (430,196,800,310)
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()
""" cv2.imshow('img',ut.resize(img))
cv2.imwrite('img.png',img)
cv2.waitKey(0)
cv2.destroyAllWindows() """
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment