mirror of
https://github.com/megvii-research/NAFNet.git
synced 2025-06-03 21:55:00 +08:00
cf36476ea3
python basicsr/demo.py -opt options/test/SIDD/NAFNet-width64.yml --input_path ./demo/noisy.png --output_path ./demo/denoise_img.png python basicsr/demo.py -opt options/test/REDS/NAFNet-width64.yml --input_path ./demo/blurry.jpg --output_path ./demo/deblur_img.png Signed-off-by: Ranjan Debnath <ranjandebnath.rd@gmail.com>
21 lines
557 B
Python
21 lines
557 B
Python
from copy import deepcopy
|
|
|
|
from basicsr.utils.registry import METRIC_REGISTRY
|
|
from .niqe import calculate_niqe
|
|
from .psnr_ssim import calculate_psnr, calculate_ssim
|
|
|
|
__all__ = ['calculate_psnr', 'calculate_ssim', 'calculate_niqe']
|
|
|
|
|
|
def calculate_metric(data, opt):
|
|
"""Calculate metric from data and options.
|
|
|
|
Args:
|
|
opt (dict): Configuration. It must contain:
|
|
type (str): Model type.
|
|
"""
|
|
opt = deepcopy(opt)
|
|
metric_type = opt.pop('type')
|
|
metric = METRIC_REGISTRY.get(metric_type)(**data, **opt)
|
|
return metric
|