python判断图片相似_Python2-比较当前图片跟图库那个图片相似

#-*- coding: utf-8 -*-

'''Created on 2018年5月17日'''

from selenium importwebdriverfrom PIL importImageimportrequestsimporttimeimportbase64importbase64importrequestsfrom urllib importurlencodeimportjson#requests.packages.urllib3.disable_warnings()

importdatetimefrom time importstrftimefrom time importsleepfrom PIL importImage#import pytesseract

from PIL importImageimportosimportcv2from numpy importaverage, dot, linalgimportheapqimportcollectionsfrom lib.readConfig importReadconfig

conf=Readconfig()

filedir=conf.getConfigValue("filedir")defgetGray(image_file):

tmpls=[]for h in range(0, image_file.size[1]):#h

for w in range(0, image_file.size[0]):#w

tmpls.append( image_file.getpixel((w,h)) )returntmplsdef getAvg(ls):#获取平均灰度值

return sum(ls)/len(ls)defgetMH(i1,i2):

a=getImgHash(i1)

b=getImgHash(i2)

dist=0;for i inrange(0,len(a)):if a[i]==b[i]:

dist=dist+1

returndistdefgetImgHash(fne):

image_file= Image.open(fne) #打开

image_file=image_file.resize((35,35))#重置图片大小我12px X 12px

image_file=image_file.convert("L")#转256灰度图

Grayls=getGray(image_file)#灰度集合

avg=getAvg(Grayls)#灰度平均值

bitls=''#接收获取0或1

for h in range(1, image_file.size[1]-1):#h

for w in range(1, image_file.size[0]-1):#w

if image_file.getpixel((w,h))>=avg:#像素的值比较平均值 大于记为1 小于记为0

bitls=bitls+'1'

else:

bitls=bitls+'0'

returnbitlsdefmatch1(a,rootdir):

list=os.listdir(rootdir)

li=[]for i inlist:#print rootdir+"/"+i

re=getMH(a,rootdir+"/"+i)

li.append(re)#print li#print max(li)

b=str(li.index(max(li))+1)returnbdefg_code(pic):

dic={"1":"2","2":"3","3":"4","4":"5","5":"6","6":"7","7":"8","8":"9","9":"a","10":"b","11":"c","12":"d","13":"e","14":"f","15":"g","16":"h","17":"i","18":"j","19":"k","20":"m","21":"n","22":"p","23":"q","24":"r","25":"s","26":"t","27":"u","28":"v","29":"w","30":"x","31":"y","32":"z"}

img=Image.open(pic)

a=img.size[0]

b=img.size[1]

p1=filedir+r'eos_tdym/lib/pic/das1.png'p2=filedir+r'eos_tdym/lib/pic/das2.png'p3=filedir+r'eos_tdym/lib/pic/das3.png'p4=filedir+r'eos_tdym/lib/pic/das4.png'dir1=filedir+r'eos_tdym/lib/pic/pic1'dir2=filedir+r'eos_tdym/lib/pic/pic2'dir3=filedir+r'eos_tdym/lib/pic/pic3'dir4=filedir+r'eos_tdym/lib/pic/pic4'cropped= img.crop((0, 0, 25, 30)) #(left, upper, right, lower)

cropped.save(p1)

cropped= img.crop((25, 0, 50, 30)) #(left, upper, right, lower)

cropped.save(p2)

cropped= img.crop((50, 0, 75, 30)) #(left, upper, right, lower)

cropped.save(p3)

cropped= img.crop((75, 0, 96, 30)) #(left, upper, right, lower)

cropped.save(p4)

re1=str(match1(p1,dir1))

re2=str(match1(p2,dir2))

re3=str(match1(p3,dir3))

re4=str(match1(p4,dir4))print u"获取到验证码:"+dic[re1]+dic[re2]+dic[re3]+dic[re4]returndic[re1],dic[re2],dic[re3],dic[re4]defg_code1(pic):

dic={"1":"2","2":"3","3":"4","4":"5","5":"6","6":"7","7":"8","8":"9","9":"a","10":"b","11":"c","12":"d","13":"e","14":"f","15":"g","16":"h","17":"i","18":"j","19":"k","20":"m","21":"n","22":"p","23":"q","24":"r","25":"s","26":"t","27":"u","28":"v","29":"w","30":"x","31":"y","32":"z"}

img=Image.open(pic)

a=img.size[0]

b=img.size[1]

p1="pic5/das1.png"p2="pic5/das2.png"p3="pic5/das3.png"p4="pic5/das4.png"dir1="pic1"dir2="pic2"dir3="pic3"dir4="pic4"cropped= img.crop((0, 0, 25, 30)) #(left, upper, right, lower)

cropped.save(p1)

cropped= img.crop((25, 0, 50, 30)) #(left, upper, right, lower)

cropped.save(p2)

cropped= img.crop((50, 0, 75, 30)) #(left, upper, right, lower)

cropped.save(p3)

cropped= img.crop((75, 0, 96, 30)) #(left, upper, right, lower)

cropped.save(p4)

re1=match1(p1,dir1)

re2=match1(p2,dir2)

re3=match1(p3,dir3)

re4=match1(p4,dir4)printdic[re1]printdic[re2]printdic[re3]printdic[re4]return dic[re1],dic[re2],dic[re3],dic[re4]