7e786cbaa2af5a666a3957b11d9621ae9f34a883,test.py,,,#,16

Before Change

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

        out = model(image)
        mse = ((target - out) ** 2).mean()
        psnr = 10 * log10(1 / mse.data.cpu().numpy())

After Change

        image = Image.open(data_path + image_name)
        image = Variable(ToTensor()(image)).unsqueeze(0)
        target = Image.open(target_path + image_name)
        target = Variable(ToTensor()(target)).unsqueeze(0)
        if torch.cuda.is_available():
            image.cuda()
            target.cuda()

Instances