Tkinter抠图软件源代码

https://github.com/siddharthm/PyBGremover/blob/master/main.py

1. # Python program to process a feed file, download all images
   
2. # and process them to remove background from them automatically
   
3. # and then using a tool to manually remove background from
   
4. # suspected images
   
5. 
   
6. from sys import argv,exit
   
7. import interior_removal as ir
   
8. import region_shrinking as rs
   
9. try:
   
10.         from PIL import Image
    
11. except:
    
12.         print "PIL not installed. Install it before proceeding"
    
13. import os
    
14. import itertools
    
15. 
    
16. # parameters for setting the color bucket
    
17. # la is the size of bucket
    
18. # plus is the amt added to color so that the colors near 255 have a filled bucket
    
19. 
    
20. la=7
    
21. plus=3
    
22. 
    
23. def toSuspect(img,omg):
    
24.         img_LA=img.convert('RGBA').convert('LA')
    
25.         removeObject(img_LA,omg)
    
26.         omg_LA=omg.convert('LA')
    
27.         bg=getbgColor(img_LA)
    
28.         X,Y=img.size
    
29.         count,count2=0,0
    
30.         pixels=omg.load()
    
31.         pixel1=omg_LA.load()
    
32.         for x,y in itertools.product(range(X),range(Y)):
    
33.                 if pixels[x,y][-1]==0:
    
34.                         continue
    
35.                 if bg==(la*int((pixel1[x,y][0]+plus)/la)):
    
36.                         count+=1
    
37.                 count2+=1
    
38.         if count>(count2/200):
    
39.                 return True
    
40.         return False
    
41.                
    
42. def removeObject(img,omg):
    
43.         pixels=img.load()
    
44.         pixels2=omg.load()
    
45.         X,Y=img.size
    
46.         for x,y in itertools.product(range(X),range(Y)):
    
47.                 if pixels2[x,y][-1]>0:
    
48.                         pixels[x,y]=(0,0)
    
49. 
    
50. def getbgColor(img):
    
51.         pixels=img.load()
    
52.         X,Y=img.size
    
53.         dict2={}
    
54.         for x,y in itertools.product(range(X),range(Y)):
    
55.                 if pixels[x,y][-1]>0:
    
56.                         key=la*int((pixels[x,y][0]+plus)/la)
    
57.                         if key not in dict2:
    
58.                                 dict2[key]=0
    
59.                         dict2[key]+=1
    
60.         return max(dict2.items(),key=lambda x:x[1])[0]
    
61. 
    
62. def main(in_folder,out_folder):
    
63.         if not os.path.exists(out_folder+"/suspect"):
    
64.                 os.makedirs(out_folder+"/suspect")
    
65.         for dirname,dirnames,filenames in os.walk(in_folder):
    
66.                 for image in filenames:
    
67.                         imagename=image.split(".")[0]
    
68.                         print "Now doing image:",image
    
69.                         img=Image.open(dirname+"/"+image)
    
70.                         if not os.path.exists(out_folder+"/"+imagename+".png") and not os.path.exists(out_folder+"/suspect/"+imagename+".png"):
    
71.                                 #img2=Image.new(mode="RGBA",size=(img.size[0]*2,img.size[1]))
    
72.                                 omg,thresc=rs.region_shrink3(img,img.size[0],img.size[1])
    
73.                                 #img2.paste(img,(0,0))
    
74.                                 #img2.paste(omg,(img.size[0],0))
    
75.                                 if toSuspect(img,omg):
    
76.                                         print "\tSuspected"
    
77.                                         omg.save(out_folder+"/suspect/"+imagename+".png","PNG")
    
78.                                 else:
    
79.                                         omg.save(out_folder+"/"+imagename+".png","PNG")
    
80. 
    
81. 
    
82. if __name__=="__main__":
    
83.         try:
    
84.                 in_folder=argv[1]
    
85.                 out_folder=argv[2]
    
86.                 if not os.path.exists(out_folder):
    
87.                         os.makedirs(out_folder)
    
88.         except:
    
89.                 print "Provide the following arguments"
    
90.                 print "1. Input folder name"
    
91.                 print "2. Output folder name"
    
92.                 exit()
    
93.         main(in_folder,out_folder)
    
94.                

复制代码