python像素级比对_python +selenium 实现图片对比功能

#encoding=utf-8

from selenium import webdriver

import unittest, time

from PIL import Image

class ImageCompare(object):

‘‘‘

本类实现了对两张图片通过像素比对的算法，获取文件的像素个数大小

然后使用循环的方式将两张图片的所有项目进行一一对比，

并计算比对结果的相似度的百分比

‘‘‘

def make_regalur_image(self, img, size=(256, 256)):

# 将图片尺寸强制重置为指定的size大小

# 然后再将其转换成RGB值

return img.resize(size).convert(‘RGB‘)

def split_image(self, img, part_size=(64, 64)):

# 将图片按给定大小切分

w, h = img.size

pw, ph = part_size

assert w % pw == h % ph == 0

return [img.crop((i, j, i + pw, j + ph)).copy() for i in range(0, w, pw) for j in range(0, h, ph)]

def hist_similar(self, lh, rh):

# 统计切分后每部分图片的相似度频率曲线

assert len(lh) == len(rh)

return sum(1 - (0 if l == r else float(abs(l - r))