python 对比图片相似度

时间：2023-03-08 19:28:13

最近appium的使用越来越广泛了，对于测试本身而言，断言同样是很重要的，没有准确的断言那么就根本就不能称之为完整的测试了。那么目前先从最简单的截图对比来看。我这里分享下python的图片相似度的代码。
目前我自己工作中全部是使用python的PIL库，绝对很赞！

#sudo pip install PIL

def pil_image_similarity(filepath1, filepath2):

    from PIL import Image

    import math

    import operator

    image1 = Image.open(filepath1)

    image2 = Image.open(filepath2)

#    image1 = get_thumbnail(img1)

#    image2 = get_thumbnail(img2)

    h1 = image1.histogram()

    h2 = image2.histogram()

    rms = math.sqrt(reduce(operator.add,  list(map(lambda a,b: (a-b)**2, h1, h2)))/len(h1) )

    return rms

print pil_image_similarity('/Users/apple/Desktop/git/Vimi_API_Test/Compare_image_test/output.jpg','/Users/apple/Desktop/git/Vimi_API_Test/Compare_image_test/0.jpg')

相关文章