0x2620/pandora
2
1
Fork 1
Code Issues 1.3k Pull requests Releases Wiki Activity

clean up get_hash method

Browse source
This commit is contained in:
rolux 2012-12-27 14:44:28 +01:00
parent 966c6954b3
commit 2cad149414
1 changed files with 9 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
pandora/sequence/extract.py
View file

@ -9,7 +9,8 @@ for pixel_index in range(64):
x, y = pixel_index % 8, int(pixel_index / 8) x, y = pixel_index % 8, int(pixel_index / 8)
ZONE_INDEX.append(int(x / 2) + int(y / 4) * 4) ZONE_INDEX.append(int(x / 2) + int(y / 4) * 4)
def get_hash(image, mode, debug=False): def get_hash(image, mode):
image_hash = 0
if mode == 'color': if mode == 'color':
# divide the image into 8 zones: # divide the image into 8 zones:
# 0 0 1 1 2 2 3 3 # 0 0 1 1 2 2 3 3
@ -21,7 +22,6 @@ def get_hash(image, mode, debug=False):
# 4 4 5 5 6 6 7 7 # 4 4 5 5 6 6 7 7
# 4 4 5 5 6 6 7 7 # 4 4 5 5 6 6 7 7
image_data = image.getdata() image_data = image.getdata()
image_hash = 0
zone_values = [] zone_values = []
for zone_index in range(8): for zone_index in range(8):
zone_values.append([]) zone_values.append([])
@ -39,16 +39,18 @@ def get_hash(image, mode, debug=False):
)) ))
image_hash += color_index * pow(2, zone_index * 8) image_hash += color_index * pow(2, zone_index * 8)
elif mode == 'shape': elif mode == 'shape':
# pixels brighter than the mean register as 1,
# pixels equal to or darker than the mean as 0
image_data = image.convert('L').getdata() image_data = image.convert('L').getdata()
image_mean = sum(image_data) / 64 image_mean = sum(image_data) / 64
image_hash = 0
for pixel_index, pixel_value in enumerate(image_data): for pixel_index, pixel_value in enumerate(image_data):
if pixel_value > image_mean: if pixel_value > image_mean:
image_hash += pow(2, pixel_index) image_hash += pow(2, pixel_index)
#return hash as 16 character hex string image_hash = hex(image_hash)[2:].upper()
h = hex(image_hash)[2:].upper() if image_hash.endswith('L'):
if h.endswith('L'): h = h[:-1] image_hash = image_hash[:-1]
return '0' * (16-len(h)) + h image_hash = '0' * (16 - len(h)) + h
return image_hash
def get_sequences(path, position=0): def get_sequences(path, position=0):
modes = ['color', 'shape'] modes = ['color', 'shape']