zghnwsc
2021-06-01 10:15
采纳率: 100%
浏览 57

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张量转换之间的问题,萌新不太懂,请问各位大佬这里应该怎么改呢?

  • 写回答
  • 关注问题
  • 收藏
  • 邀请回答

4条回答 默认 最新

  • AlanMax 2021-06-01 10:51
    已采纳

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

    打赏 评论
  • CSDN专家-黄老师 2021-06-01 10:26

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

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

    打赏 评论
  • CSDN专家-AlanMax 2021-06-01 18:35

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

    打赏 评论
  • 有问必答小助手 2021-06-01 16:53

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

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

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

    打赏 评论

相关推荐 更多相似问题