pytorch中计算灰度图的psnr

######这里是计算psnr方式
def calc_psnr(img1,img2):   
    img1 = Variable( img1, requires_grad=False)    
    img2 = Variable( img2, requires_grad = False)
    img1 = img1.cpu()
    img1 = np.array(img1).astype(np.float32)

    img1 = torch.from_numpy(img1).float().unsqueeze(0)
   
    img2 = img2.cpu()
    img2 = np.array(img2).astype(np.float32)
   
    img2 = torch.from_numpy(img2).float().unsqueeze(0)

    PSNR = peak_signal_noise_ratio(img1,img2).item()

    return PSNR



#####提示错误
AttributeError: 'torch.dtype' object has no attribute 'type'




########这里是前面训练过程中调用psnr更新，不会出错
epoch_psnr.update(calc_psnr(preds, labels), len(inputs))

我在训练神经网络时，需要用灰度图数据集，使用传统的RGB转ycbcr的方法，psnr会输出nan。使用from skimage.metrics import peak_signal_noise_ratio计算的话，经过测试两幅图片可以计算。但是在训练过程中，给对应的preds和lable训练时，会出现numpy数组很tensor张量转换之间的问题，萌新不太懂，请问各位大佬这里应该怎么改呢？

img1,img2从gpu拿到cpu还是tensor，要img.cpu().numpy(),然后进行格式转换。如果有帮助请采纳。

torch.dtype没有属性type，你检测一下代码是不是写错了，如果没错，确认一下pytorch的版本，版本不同会导致相应的函数方法不太一样

如果对你有帮助，可以点击我这个回答右上方的【采纳】按钮，给我个采纳吗，谢谢

你好，我换了个新的账号来帮你解决问题，可以把完整代码或者报错私信发我帮你解决下问题嘛

您好，我是有问必答小助手，您的问题已经有小伙伴解答了，您看下是否解决，可以追评进行沟通哦~

如果有您比较满意的答案 / 帮您提供解决思路的答案，可以点击【采纳】按钮，给回答的小伙伴一些鼓励哦～～

ps:问答VIP仅需29元，即可享受5次/月 有问必答服务，了解详情>>>https://vip.csdn.net/askvip?utm_source=1146287632