python 圖片搜尋

#!/usr/bin/python# -*- coding: cp936 -*-import globimport osimport sysfrom PIL import ImageEXTS = 'jpg', 'jpeg', 'JPG', 'JPEG', 'gif', 'GIF', 'png', 'PNG'file_object = open('ͼƬHashֵ.txt', 'w')def avhash(im):    if not isinstance(im, Image.Image):        im = Image.open(im)    im = im.resize((8, 8), Image.ANTIALIAS).convert('L')    avg = reduce(lambda x, y: x + y, im.getdata()) / 64.    return reduce(lambda x, (y, z): x | (z << y),                  enumerate(map(lambda i: 0 if i < avg else 1, im.getdata())),                  0)def hamming(h1, h2):    h, d = 0, h1 ^ h2    while d:        h += 1        d &= d - 1    return hif __name__ == '__main__':        #im, wd = sys.argv[1], '.' if len(sys.argv) < 3 else sys.argv[2]        im = r'psu (1).jpg'        wd = r'C:\tupian_ku'        h = avhash(im)        os.chdir(wd)        images = []        for ext in EXTS:            images.extend(glob.glob('*.%s' % ext))        seq = []        prog = int(len(images) > 50 and sys.stdout.isatty())        for f in images:            seq.append((f, hamming(avhash(f), h)))            if prog:                perc = 100. * prog / len(images)                x = int(2 * perc / 5)                print '\rCalculating... [' + '#' * x + ' ' * (40 - x) + ']',                print '%.2f%%' % perc, '(%d/%d)' % (prog, len(images)),                sys.stdout.flush()                prog += 1        if prog: print        for f, ham in sorted(seq, key=lambda i: i[1]):            print "%d\t%s" % (ham, f)            file_object.write("%d\t%s\n" % (ham, f))file_object.close( )