ssim计算问题

[kerryhhh]

Invertible-Image-Rescaling/codes/utils/util.py

Lines 167 to 185 in 93aaa94

	def calculate_ssim(img1, img2):
	'''calculate SSIM
	the same outputs as MATLAB's
	img1, img2: [0, 255]
	'''
	if not img1.shape == img2.shape:
	raise ValueError('Input images must have the same dimensions.')
	if img1.ndim == 2:
	return ssim(img1, img2)
	elif img1.ndim == 3:
	if img1.shape[2] == 3:
	ssims = []
	for i in range(3):
	ssims.append(ssim(img1, img2))
	return np.array(ssims).mean()
	elif img1.shape[2] == 1:
	return ssim(np.squeeze(img1), np.squeeze(img2))
	else:
	raise ValueError('Wrong input image dimensions.')

计算ssim这里的第180行是不是写错了？处理3通道图片的时候

[pkuxmq]

是的，应该是ssims.append(ssim(img1[..., i], img2[..., i]))，这一部分的代码是直接使用的当时BasicSR中的代码，后来BasicSR应该对此进行了修正。