python对比两组以mat格式数组形式表示的图片，得到SSIM

利用python, 两组图片数据均以mat格式表示，每组包含多个图片数组，同时得到这几对图片的SSIM。

import matplotlib.pyplot as plt
import numpy as np
import scipy.io as sio
from skimage.measure import compare_ssim as ssim
import os
# 定义mat文档中的图片矩阵个数与shape
data1 = np.zeros(shape=(5, 100, 100))
data2 = np.zeros(shape=(5, 100, 100))

mat_filename_list = os.listdir("E:/SSIM对比/data1")

for index, mat_filename in enumerate(mat_filename_list):
    matdata1 = sio.loadmat("E:/SSIM对比/data1/" + mat_filename)
    data1[index] = matdata1["data"]

print(np.array(data1).shape)
data1 = np.array(data1)

mat_filename_list = os.listdir("E:/SSIM对比/data2")

for index, mat_filename in enumerate(mat_filename_list):
    matdata2 = sio.loadmat("E:/SSIM对比/data2/" + mat_filename)
    data2[index] = matdata2["data"]

print(np.array(data2).shape)
data2 = np.array(data2)

for n in range(5):
    s = ssim(data1[n], data2[n])
    plt.imshow(data1[n])
    plt.show()
    plt.imshow(data2[n])
    plt.show()
    print(s)