好麻烦的抠图

JAY饭

昨天晚上看了一个文字游戏帖子，心血来潮想弄个pygame版，但是找不到合适的png图片只好作罢，这让我开始产生了自己抠图的想法，其实自己本身也学过两个多月的PS，但是还是喜欢自己动手写代码来完成。可惜好的算法根本看不懂，好难受，只能用粗略的算法粗糙的抠下图，而且也无法普及成其他类型的抠图
过程

JAY饭

还是放出代码吧

1. import pygame
   
2. from pygame.locals import *
   
3. import sys
   
4. 
   
5. bg = 32, 141, 240
   
6. size = w,h = 950,1150
   
7. 
   
8. screen = pygame.display.set_mode(size,RESIZABLE)
   
9. pygame.display.set_caption('抠图')
   
10. clock = pygame.time.Clock()
    
11. img = pygame.image.load('11.png')
    
12. pos = img.get_rect()
    
13. pos[0],pos[1] = 0,-400
    
14. 
    
15. temp = [(1,1),(-1,-1),(1,-1),(-1,1),(0,1),(1,0),(0,-1),(-1,0)]
    
16. lst = []
    
17. color = 600
    
18. 
    
19. def diffuse(a,b):
    
20.     global lst,color
    
21.     lst= [(a,b)]
    
22.     if sum(img.get_at((a,b))[:3])>color:
    
23.         img.set_at((a,b),[2,2,2,255])
    
24.     while True:
    
25.         a,b = lst.pop(0)
    
26.         for x,y in temp:
    
27.             a1,b1 = a+x,b+y
    
28.             if 0<=a1<949 and 0<=b1<1149:
    
29.                 if sum(img.get_at((a1,b1))[:3])>color:
    
30.                     img.set_at((a1,b1),[2,2,2,255])
    
31.                     lst.append((a1,b1))
    
32.         if len(lst)==0:
    
33.             break
    
34. 
    
35. while True:
    
36.     for e in pygame.event.get():
    
37.         if e.type == pygame.QUIT:
    
38.             sys.exit()
    
39.         elif e.type == MOUSEBUTTONDOWN:
    
40.             if e.button == 1:
    
41.                 a,b = pygame.mouse.get_pos()
    
42.                 print(a,b+400)
    
43.                 print(img.get_at((a,b+400)))
    
44.                 diffuse(a,b+400)
    
45.             if e.button == 2:
    
46.                 a,b = pygame.mouse.get_pos()
    
47.                 color = sum(img.get_at((a,b+400))[:3])-5
    
48.                 print(color)
    
49.                 print(img.get_at((a,b+400)))
    
50.         elif e.type == KEYDOWN:
    
51.             if e.key == K_a:
    
52.                 for i in range(950):
    
53.                     for j in range(1150):
    
54.                         if img.get_at((i,j))[3] == 0:
    
55.                             img.set_at((i,j),(32, 141, 240,255))
    
56.                 pygame.image.save(img,'副本1.png')
    
57.                
    
58.     screen.fill(bg)
    
59.     screen.blit(img,pos)
    
60.     pygame.display.flip()
    
61.     clock.tick(40)
    
62. 
    

复制代码

第二段是，直线型自动判断噪点，也想过用广度扩散的方法自动判断，但是会不可避免的影响被抠的图本身，就算了。

1. from PIL import Image
   
2. import numpy as np
   
3. 
   
4. img = Image.open('11.pNg').convert('RGBA')
   
5. w,h = img.size
   
6. data = np.reshape(img,(h,w,4))
   
7. new_data = data.copy()
   
8. print(new_data[0,0,0:3])
   
9. 
   
10. temp = [(0,1),[1,0],(-1,0),(0,-1)]
    
11. 
    
12. for x in range(h):
    
13.     for y in range(w):
    
14.         if new_data[x,y,0:3].sum() == 6:
    
15.                 new_data[x,y,3] = 0
    
16.                
    
17. for x in range(h):
    
18.     for y in range(w):
    
19.         if new_data[x,y,3] !=0:
    
20.             if new_data[x,y,0:3].sum() > 530:
    
21.                 new_data[x,y,3] = 0
    
22.             else:
    
23.                 break
    
24. for x in range(h-1,0,-1):
    
25.     for y in range(w-1,0,-1):
    
26.         if new_data[x,y,3] !=0:
    
27.             if new_data[x,y,0:3].sum() > 530:
    
28.                 new_data[x,y,3] = 0
    
29.             else:
    
30.                 break
    
31. 
    
32. for x in range(h):
    
33.     for y in range(w-1,0,-1):
    
34.         if new_data[x,y,3] !=0:
    
35.             if new_data[x,y,0:3].sum() > 490:
    
36.                 new_data[x,y,3] = 0
    
37.             else:
    
38.                 break
    
39. 
    
40. 
    
41. newimg = Image.fromarray(new_data)
    
42. newimg.save('timg2.png','PNG')

复制代码

JAY饭

第二张是半成品，最后一张是PNG格式，也就是透明的，但是效果还是很不好，反正就是没有渲染效果，可惜以我现在的水平，这个效果只能说达到自己的上限。贝叶斯方法的抠图奈何真的看不懂，很难受，代码就不放了，很渣，思路是这样的，一个是pygame的选择目标点，扩散式染色，一个是用image模块，从左右上下四个方向修图。

重新改了下图片，把第三张合成了，这样可以看到明显的缺点

BngThea

牛逼

JAY饭

BngThea 发表于 2018-4-7 14:49
牛逼

大神你真的过奖，很简单的思路

JAY饭

但是对付一些简单的图片还是绰绰有余，第二张是处理后的png,比如

清风揽月shine

新手·ing

大神

小甲鱼

哟哟哟，帅哦~

JAY饭

小甲鱼 发表于 2018-4-8 01:12
哟哟哟，帅哦~

谢谢鼓励，惊了，自我感觉写的很差劲的。
       对了，上面的代码是我后期改动了的，随手贴上去的部分，所以，并不能实现完整抠图过程，我觉得像我写的这个代码也没什么好看的，看不看都不重要，最重要的还是贯穿解决问题的逻辑思路。多动脑~