compare two image similarity

compare_image_similarity.sh

# -*- coding:utf-8 -*-
# Created by shellbye on 2017/9/19.

from PIL import Image
from skimage.measure import compare_ssim as ssim
import numpy as np

# requirements
# pip install scikit-image
# pip install pillow

def load_image(infilename):
    img = Image.open(infilename)
    img.load()
    data = np.asarray(img, dtype="int32")
    return data


if __name__ == '__main__':
    imageA = load_image('images/1.jpg')
    imageB = load_image('images/2.jpg')
    s = ssim(imageA, imageB, multichannel=True)
    print(s)