IQA-PyTorch快速使用教程

胡说八道的Dr. Zhu

已于 2024-12-04 09:41:55 修改

阅读量1.8k

点赞数 39

文章标签： pytorch 人工智能 python

于 2024-11-21 11:05:41 首次发布

本文链接：https://blog.youkuaiyun.com/zly19980718/article/details/143930970

版权

官方GitHub地址

pyiqa0.1.13官方文档

关于PSNR、SSIM、FID的简要介绍可以参考这篇博客：图像修复算法常用评估指标介绍及Python代码（PSNR/SSIM/FID）-优快云博客

1、pip安装稳定版

pip install pyiqa

安装的版本是pyiqa==0.1.13（2024年11月）。

2、简单了解

import pyiqa
import torch

# list all available metrics
print(pyiqa.list_models())

输出结果为：

['ahiq', 'arniqa', 'arniqa-clive', 'arniqa-csiq', 'arniqa-flive', 'arniqa-kadid', 'arniqa-live', 'arniqa-spaq', 'arniqa-tid', 'brisque', 'brisque_matlab', 'ckdn', 'clipiqa', 'clipiqa+', 'clipiqa+_rn50_512', 'clipiqa+_vitL14_512', 'clipscore', 'cnniqa', 'cw_ssim', 'dbcnn', 'dists', 'entropy', 'fid', 'fsim', 'gmsd', 'hyperiqa', 'ilniqe', 'inception_score', 'laion_aes', 'liqe', 'liqe_mix&