python图片相似度比对源代码
时间: 2023-07-17 18:09:34 浏览: 123
以下是一个使用Python进行图片相似度比对的示例代码:
```python
from PIL import Image
import imagehash
def calculate_similarity(image1_path, image2_path):
# 加载图像
image1 = Image.open(image1_path)
image2 = Image.open(image2_path)
# 计算图像的hash值
hash1 = imagehash.average_hash(image1)
hash2 = imagehash.average_hash(image2)
# 计算汉明距离(Hamming distance)
# 汉明距离越小,表示图像越相似
similarity = 1 - (hash1 - hash2) / len(hash1.hash) ** 2
return similarity
# 测试示例
image1_path = 'image1.jpg'
image2_path = 'image2.jpg'
similarity = calculate_similarity(image1_path, image2_path)
print(f"The similarity between the two images is: {similarity}")
```
上述代码使用了Pillow库(PIL)来加载图像,并使用imagehash库计算图像的平均哈希(average hash)。然后,通过计算汉明距离来衡量图像的相似度。汉明距离越小,表示图像越相似。
你可以将`image1.jpg`和`image2.jpg`替换为你要比较的两个图像的文件路径。最后,代码将打印出两个图像之间的相似度。
阅读全文