d8ca380c2d56fdbb21ec2a2440a7ca74689596b7,test.py,,,#,77

Before Change

            lr_img = tensor2im(data["input"], mean, stddev)
            if args.eval:
                hr_img = tensor2im(
                    data.get("target", data["raw_input"]), mean, stddev)
                print_evaluation(
                    osp.basename(data["input_fn"][0]),
                    *eval_psnr_and_ssim(sr_img, hr_img, args.upscale_factor))

After Change

    model = cls_model(**checkpoint["params"]["G"])
    model.load_state_dict(checkpoint["state_dict"])

    info("Phase: {}".format(config.phase.TEST))
    info("Checkpoint: {}".format(osp.basename(args.checkpoint)))

    model.eval()

Instances