看不懂用cv2.grabcut()方法抠图

JAY饭

       原文在这里https://blog.csdn.net/u010682375/article/details/72791864
        里面第二次提到的掩码图片是哪一种呢，我试了几种都失败了，尤其是下面这段：
       
        ‘’实际上我是怎么做的呢？我们使用图像编辑软件打开输入图像，添加一个图层，使用笔刷工具在需要的地方使用白色绘制（比如头发，鞋子，球等）；使用黑色笔刷在不需要的地方绘制                      （比如， logo，草地等）。然后将其他地方用灰色填充，保存成新的掩码图像。在 OpenCV 中导入这个掩模图像，根据新的
掩码图像对原来的掩模图像进行编辑。代码如下：

newmask = cv2.imread('newmask.jpg',0)
mask[newmask == 0] = 0
mask[newmask == 255] = 1
mask,bgdModel,fgdModel = cv2.grabCut(img, mask, None, bgdModel, fgdModel, 5, cv2.GC_INIT_WITH_MASK)
mask = np.where((mask ==2)|(mask ==0),0,1).astype('uint8') ‘’
      
        上面这段里面他这个‘newmask.jpg’到底是上面生成的 半成品抠图 ，还是这个半成品抠图上面的被手工标注的图层，还有里面说到的0像素，1像素真的不太明白他到底说的什么意思。
像素不都是0，0，0，0这种格式吗

下面发我实现的效果，始终不懂怎么去掉非前景部分，依次是，原图，第一次原始掩码的抠图，我用自己写的蒙版代码画的前景，背景蒙版图层，然后将其代入第二次掩码，最后是最后显示的图，效果不理想。

JAY饭

抱歉忘了发悬赏，需要的话我在后面补，求大神指导下

JAY饭

老实说很头疼这种复制 别人翻译的国外文章却又不求甚解的博客，讲不讲清楚，搜遍全网，得不到关于python实现的完整过程。

JAY饭

主代码，是我copy的，改动了一些，但不知道改的对不对，我真的不能理解它
，看不懂

1. import cv2
   
2. import numpy as np
   
3. from matplotlib import pyplot as plt
   
4. 
   
5. img = cv2.imread('timg.jpg')
   
6. mask = np.zeros(img.shape[:2],np.uint8)
   
7. bgdModel = np.zeros((1,65),np.float64)
   
8. fgdModel = np.zeros((1,65),np.float64)
   
9. rect = (100,10,415,471)
   
10. cv2.grabCut(img, mask, rect, bgdModel, fgdModel,5,cv2.GC_INIT_WITH_RECT)
    
11. mask2 = np.where((mask ==2)|(mask==0),0,1).astype('uint8')
    
12. img = img*mask2[:,:,np.newaxis]
    
13. 
    
14. cv2.imwrite('3.jpg',img)#这里讲第一次抠图保存
    
15. 
    
16. 
    
17. newmask = cv2.imread('5.png',0)#这里将上面得到的图片，用蒙版处理后的图层，也就是图3
    
18. 
    
19. # whereever it is marked white (sure foreground), change mask=1
    
20. # whereever it is marked black (sure background), change mask=0
    
21. mask[newmask == 0] = 0
    
22. mask[newmask == 255] = 1
    
23. 
    
24. mask, bgdModel, fgdModel = cv2.grabCut(img,mask,None,bgdModel,fgdModel,5,cv2.GC_INIT_WITH_MASK)
    
25. 
    
26. mask = np.where((mask==2)|(mask==0),0,1).astype('uint8')
    
27. img = img*mask[:,:,np.newaxis]
    
28. plt.imshow(img),plt.colorbar(),plt.show()
    
29. 
    

复制代码

我自己写的绘制蒙版图层代码，随便划拉的几行，没有加工，因为这个不是重点

1. import pygame
   
2. from pygame.locals import *
   
3. import sys
   
4. 
   
5. bg = 255,255,255
   
6. bg1 = 176,196,22
   
7. size = width,hright = 516,481
   
8. screen =pygame.display.set_mode(size)
   
9. clock = pygame.time.Clock()
   
10. 
    
11. img = pygame.image.load('3.jpg').convert_alpha()
    
12. img1 = pygame.Surface(size)
    
13. img1.fill(bg1)
    
14. img1.set_alpha(100)
    
15. pos = img.get_rect()
    
16. 
    
17. w= False
    
18. w1= False
    
19. 
    
20. def write():
    
21.     global w,w1
    
22.     if w:
    
23.         x,y = pygame.mouse.get_pos()
    
24.         img1.set_at((x,y),(255,255,255,255))
    
25.         img1.set_at((x+1,y),(255,255,255,255))
    
26.         img1.set_at((x-1,y),(255,255,255,255))
    
27.         img1.set_at((x+2,y),(255,255,255,255))
    
28.         img1.set_at((x-2,y),(255,255,255,255))
    
29.         img1.set_at((x,y+1),(255,255,255,255))
    
30.         img1.set_at((x,y+2),(255,255,255,255))
    
31.         img1.set_at((x,y-1),(255,255,255,255))
    
32.         img1.set_at((x,y-2),(255,255,255,255))
    
33.     if w1:
    
34.         x,y = pygame.mouse.get_pos()
    
35.         img1.set_at((x,y),(0,0,0,255))
    
36.         img1.set_at((x+1,y),(0,0,0,255))
    
37.         img1.set_at((x-1,y),(0,0,0,255))
    
38. while True:
    
39.     for e in pygame.event.get():
    
40.         if e.type == pygame.QUIT:
    
41.             sys.exit()
    
42.         if e.type == MOUSEBUTTONDOWN:
    
43.             if e.button == 1:
    
44.                 w = True
    
45.             elif e.button == 3:
    
46.                 w1= True
    
47.                
    
48.         if e.type == MOUSEBUTTONUP:
    
49.             if e.button == 1:
    
50.                 w = False
    
51.             elif e.button == 3:
    
52.                 w1 = False
    
53. 
    
54.         if e.type == KEYDOWN:
    
55.             if e.key == K_a:
    
56.                 img1.set_alpha(225)
    
57.                 pygame.image.save(img1,'5.png')
    
58.                
    
59.     screen.fill(bg)
    
60.     screen.blit(img,pos)
    
61.     screen.blit(img1,pos)
    
62.     write()
    
63.     pygame.display.flip()
    
64.     clock.tick(200)

复制代码

JAY饭

已解决~

casanava

请教大侠，修复模糊图片用什么函数