a
    d-                     @   sH   d dl mZ d dlmZ ddlmZ ddlmZmZ g dZ	dd Z
d	S )
    )deepcopy)METRIC_REGISTRY   )calculate_niqe)calculate_psnrcalculate_ssim)r   r   r   c                 C   s.   t |}|d}t|f i | |}|S )zCalculate metric from data and options.

    Args:
        opt (dict): Configuration. It must contain:
            type (str): Model type.
    type)r   popr   get)dataoptZmetric_typeZmetric r   a/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/basicsr/metrics/__init__.pycalculate_metric
   s    
r   N)copyr   Zbasicsr.utils.registryr   Zniqer   Z	psnr_ssimr   r   __all__r   r   r   r   r   <module>   s
   