Python中的结构相似指数(SSIM) (多通道错误)

Question

我想要计算生成的图像和目标图像之间的结构相似性指数(SSIM) (从图像数组中随机选取)。

这就是我试过的-

from skimage.metrics import structural_similarity as ssim

print(tar_image.shape)
print(gen_image.shape)

ssim_skimg = ssim(tar_image, gen_image,
                  data_range = gen_image.max() - gen_image.min(), 
                  multichannel = True)

print("SSIM: based on scikit-image = ", ssim_skimg)

但是我得到了这个输出：

(1, 128, 128, 3)
(1, 128, 128, 3)

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

有谁能告诉我我哪里出了问题，以及如何解决这个问题？

Answer 1

您有3个通道图像，因此应该使用multichannel = True参数。

此外，您还应该删除图像的第一个维度，以获得(128,128,3)形状。

import numpy as np
from skimage.metrics import structural_similarity as ssim 

tar_image = np.zeros((128, 128, 3))
gen_image = np.zeros((128, 128, 3))

ssim_skimg = ssim(tar_image, gen_image, multichannel = True)