问题描述 在用skimage的函数计算两张图片的SSIM时: from skimage.metrics import structural_similarity as SSIM 我想在模型训练过程中计算验证集GT和pre图片的SSIM,对输入SSIM函数的图像shape已转换为(B,H,W,C): SSIM((data_val[1].numpy().transpose(0, 2, 3, 1)*255).astype(np.uint8