code来源:https://github.com/JohannesBuchner/imagehash 外文原文:https://fullstackml.com/wavelet-image-hash-in-python-3504fdd282b5
可以直接pip:
pip install imagehash
感知哈希,不同于aHash,但首先它确实是离散余弦变换和频域。
主函数:def phash(image, hash_size=8, highfreq_factor=4):
案例:
highfreq_factor = 1
hash_size = 8
img_size = hash_size * highfreq_factor
hash1 = imagehash.phash(Image.open('1_1.jpg'),hash_size=hash_size,highfreq_factor=highfreq_factor)
print(hash1)
# > 354adab5054af0b7
hash2 = imagehash.phash(Image.open('5_1.jpg'),hash_size=hash_size,highfreq_factor=highfreq_factor)
print(hash2)
# > 5b7724c8bb364551
1 - (hash1 - hash2)/len(hash1.hash)**2 # 相似性
平均散列,对于每个像素输出1,如果该像素是大于或等于平均值,否则为0。 主函数:
average_hash(image, hash_size=8)
案例:
hash_size = 6
hash1 = imagehash.average_hash(Image.open('1_1.jpg'),hash_size=hash_size)
print(hash1)
# > 354adab5054af0b7
hash2 = imagehash.average_hash(Image.open('5_1.jpg'),hash_size=hash_size)
print(hash2)
# > 5b7724c8bb364551
1 - (hash1 - hash2)/len(hash1.hash)**2 # 相似性
梯度散列,计算每个像素的差值,并与平均差异的差异进行比较。
def dhash(image, hash_size=8)
案例:
hash_size = 10
hash1 = imagehash.dhash(Image.open('5_1.jpg'),hash_size=hash_size)
print(hash1)
# > 354adab5054af0b7
hash2 = imagehash.dhash(Image.open('1_1.jpg'),hash_size=hash_size)
print(hash2)
# > 5b7724c8bb364551
1 - (hash1 - hash2)/len(hash1.hash)**2 # 相似性
离散小波变换(DWT)是频表示的另一种形式。流行的DCT和傅立叶变换使用余弦函数作为sin\cos的基础:sin(x),sin(2x),sin(3x)等等。与此相反,DWT使用一个单一的功能作为基础,但在不同的形式:缩放和移动。基础功能是可以改变的,这就是为什么我们可以有Haar小波,Daubechie-4小波等,这尺度效应给我们很大“时频表示”的时候,低频部分类似于原始信号。
小波散列,几天前我把它添加到库里。它的工作原理在频域中作为pHash但它使用DWT代替DCT变换。 主函数:
def whash(image, hash_size = 8, image_scale = None, mode = 'haar', remove_max_haar_ll = True)
参数:
案例:
hash_size = 8
mode = 'db4'
image_scale = 64
hash1 = imagehash.whash(Image.open('1_1.jpg'),image_scale=image_scale,hash_size=hash_size,mode = mode)
print(hash1)
# > 354adab5054af0b7
hash2 = imagehash.whash(Image.open('5_1.jpg'),image_scale=image_scale,hash_size=hash_size,mode = mode)
print(hash2)
# > 5b7724c8bb364551
1 - (hash1 - hash2)/len(hash1.hash)**2 # 相似性