在 python 中比较两个图像时，我想获得 SSIM答案

【问题标题】：I want to get the SSIM when comparing two images in python在 python 中比较两个图像时，我想获得 SSIM
【发布时间】：2020-08-18 16:35:42
【问题描述】：

我正在尝试使用“compare_ssim”功能。我目前有两个 x,y 坐标的 2xN 矩阵，其中第一行是所有 x 坐标，第二行是两个图像中每一个的所有 y 坐标。我如何计算这两张图片的 SSIM（如果有办法的话）

例如我有：

X = np.array([[1,2,3], [4,5,6]])
Y = np.array([[3,4,5],[5,6,7]])

compare_ssim(X,Y)

但我得到了错误

ValueError: win_size exceeds image extent.  If the input is a multichannel (color) image, set multichannel=True.

我不确定我是否遗漏了一个参数，或者我是否应该以这种函数工作的方式转换矩阵。或者，如果有一种方法可以将我的坐标转换为灰度矩阵？我对函数参数的矩阵应该是什么样子有点困惑。我知道它们应该是 ndarray，但 type(Y) 和 type(Y) 都是 numpy.ndarray。

【问题讨论】：

标签： python ssim

【解决方案1】：

由于您没有提到您使用的是哪个框架/库，我假设您使用的是 skimage 的 compare_ssim。

有问题的错误是由于输入的形状。您可以找到更多详细信息here。

TL;DR： compare_ssim 需要 (H, W, C) 尺寸的图像，但您的输入图像的尺寸为 (2, 3)。因此，函数混淆了将哪个维度视为通道维度。当multichannel=True时，最后一个维度被视为通道维度。

您的代码存在 3 个关键问题，

compare_image 需要 Images 作为输入。所以你的 X 和 Y 矩阵的尺寸应该是 (H, W, C) 而不是 (2, 3)
它们应该是浮点数据类型。

下面我展示了一些演示代码（注意：从 skimage v1.7 开始，compare_ssim 已移至 skimage.metrics.structural_similarity）


import numpy as np  
from skimage.metrics import structural_similarity

img1 = np.random.randint(0, 255, size=(200, 200, 3)).astype(np.float32)
img2 = np.random.randint(0, 255, size=(200, 200, 3)).astype(np.float32)

ssim_score = structural_similarity(img1, img2, multichannel=True) #score: 0.0018769083894301646

ssim_score = structural_similarity(img1, img1, multichannel=True) #score: 1.0

【讨论】：