kit/metrics/__init__.py

from .distributional import compute_fid
from .image import (
    compute_mse,
    compute_psnr,
    compute_ssim,
    compute_nmi,
    compute_mse_repeated,
    compute_psnr_repeated,
    compute_ssim_repeated,
    compute_nmi_repeated,
    compute_image_distance_repeated,
)
from .perceptual import (
    load_perceptual_models,
    compute_lpips,
    compute_lpips_repeated,
    compute_perceptual_metric_repeated,
)
from .aesthetics import (
    load_aesthetics_and_artifacts_models,
    compute_aesthetics_and_artifacts_scores,
)
from .clip import load_open_clip_model_preprocess_and_tokenizer, compute_clip_score
from .prompt import (
    load_perplexity_model_and_tokenizer,
    compute_prompt_perplexity,
)