Minor update in grabcut.py

samples/python/grabcut.py

@@ -11,10 +11,10 @@ USAGE:
 README FIRST:
     Two windows will show up, one for input and one for output.
 
-    At first, in input window, draw a rectangle around the object using
-mouse right button. Then press 'n' to segment the object (once or a few times)
+    At first, in input window, draw a rectangle around the object using the
+right mouse button. Then press 'n' to segment the object (once or a few times)
 For any finer touch-ups, you can press any of the keys below and draw lines on
-the areas you want. Then again press 'n' for updating the output.
+the areas you want. Then again press 'n' to update the output.
 
 Key '0' - To select areas of sure background
 Key '1' - To select areas of sure foreground
@@ -44,8 +44,8 @@ class App():
 
     DRAW_BG = {'color' : BLACK, 'val' : 0}
     DRAW_FG = {'color' : WHITE, 'val' : 1}
-    DRAW_PR_FG = {'color' : GREEN, 'val' : 3}
     DRAW_PR_BG = {'color' : RED, 'val' : 2}
+    DRAW_PR_FG = {'color' : GREEN, 'val' : 3}
 
     # setting up flags
     rect = (0,0,1,1)
@@ -160,14 +160,12 @@ class App():
                 print(""" For finer touchups, mark foreground and background after pressing keys 0-3
                 and again press 'n' \n""")
                 try:
+                    bgdmodel = np.zeros((1, 65), np.float64)
+                    fgdmodel = np.zeros((1, 65), np.float64)
                     if (self.rect_or_mask == 0):         # grabcut with rect
-                        bgdmodel = np.zeros((1, 65), np.float64)
-                        fgdmodel = np.zeros((1, 65), np.float64)
                         cv.grabCut(self.img2, self.mask, self.rect, bgdmodel, fgdmodel, 1, cv.GC_INIT_WITH_RECT)
                         self.rect_or_mask = 1
-                    elif self.rect_or_mask == 1:         # grabcut with mask
-                        bgdmodel = np.zeros((1, 65), np.float64)
-                        fgdmodel = np.zeros((1, 65), np.float64)
+                    elif (self.rect_or_mask == 1):       # grabcut with mask
                         cv.grabCut(self.img2, self.mask, self.rect, bgdmodel, fgdmodel, 1, cv.GC_INIT_WITH_MASK)
                 except:
                     import traceback