1433d4c435a253e891009b08a9280bc452087db0,test.py,,,#,16

Before Change

        out = model(image.unsqueeze(0))[0]
        mse = ((target - out) ** 2).mean()
        psnr = 10 * log10(1 / mse.data.cpu().numpy())
        ssim = compare_ssim(out.data.cpu().numpy(), target.data.cpu().numpy(), multichannel=True)
        out_img = ToPILImage()(out.data)
        out_img.save(out_path + "psnr_%.4f_ssim_%.4f_" % (psnr, ssim) + image_name)

After Change

        target = Variable(ToTensor()(target))
        if torch.cuda.is_available():
            image = image.unsqueeze(0).cuda()
            target = target.unsqueeze().cuda()

        out = model(image)
        mse = ((target - out) ** 2).mean()

Instances