使用python和GPU计算SSIM。

from SSIM_PIL import compare_ssim
from PIL import Image
import os
PATH = r'G:\lyb\CACD2000'
count = 0
for root, dir, file in os.walk(PATH):
    # print(root, dir, file)
    for i in file:

        image1 = Image.open(os.path.join(root, i))
        image2 = Image.open(r'E:\wzqpicture\o1.jpg')




        #  Compare images using CPU-only version
        value = compare_ssim(image1, image2, GPU=True)


        # img = Image.open(os.path.join(root, i))
        #
        # img1 = np.array(img)
        # img2 = np.array(Image.open(r'E:\wzqpicture\o1.jpg'))

        # s = ssim(img1, img2, multichannel=True, channel_axis=-1)
        count = count + 1
        if count % 100 == 0:
            print(count)
        if value > 0.6:
            img_name = i

            print(img_name)

出处：SSIM-PIL ·皮皮 (pypi.org)