35 lines
1021 B
Python
35 lines
1021 B
Python
|
from PIL import Image
|
|||
|
|
import io
|
|||
|
|
|
|||
|
|
|
|||
|
|
def compare_images(image1, image2):
|
|||
|
|
|
|||
|
|
with open(image1, 'rb') as file:
|
|||
|
|
image1_data = file.read()
|
|||
|
|
|
|||
|
|
with open(image2, 'rb') as file:
|
|||
|
|
image2_data = file.read()
|
|||
|
|
img1 = Image.open(io.BytesIO(image1_data))
|
|||
|
|
img2 = Image.open(io.BytesIO(image2_data))
|
|||
|
|
print("这里可以输出图片的相关属性，如大小、格式等", img1)
|
|||
|
|
if img1.size != img2.size:
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
pairs = zip(img1.getdata(), img2.getdata())
|
|||
|
|
print(img1.getdata())
|
|||
|
|
if len(img1.getbands()) == 1:
|
|||
|
|
# for grayscale images
|
|||
|
|
dif = sum(abs(p1 - p2) for p1, p2 in pairs)
|
|||
|
|
else:
|
|||
|
|
# for RGB images
|
|||
|
|
dif = sum(abs(c1 - c2) for p1, p2 in pairs for c1, c2 in zip(p1, p2))
|
|||
|
|
|
|||
|
|
ncomponents = img1.size[0] * img1.size[1] * 3
|
|||
|
|
similarity = (ncomponents - dif) / ncomponents
|
|||
|
|
|
|||
|
|
# 设置阈值，即两张图片相似度严格达到99%才认为它们完全一样
|
|||
|
|
return similarity > 0.99
|
|||
|
|
|
|||
|
|
|
|||
|
|
compare_images("crop_pic.png", "crop_pic.png")
|