Interactive implementation of OpenCV GrabCut
from grabcut3 import GrabCut
#Create a GrabCut object
gc = GrabCut()
gc.load_image('avengers.jpg')
gc.matte()Once you have executed the matte() method there are several options available:
r: reset the image and all grabcut parameters
n: run grabcut algorithm to matte the image
s: save the results
e: exit
0: mark background
1: mark foreground
2: mark probable background
3: mark probable foreground
+: increase size of brush
-: decrease size of brush
c: Crop ROI
You can also specify the name of the resulting image and mask, and its output directory
gc.output_params(out_dir='Output/Directory/', mask_name='MaskName0', img_name='ImageName0')