【IQA】PYIQA图像质量评估工具安装及运用

神遁克里苏

已于 2024-03-25 17:15:43 修改

阅读量3.2k

点赞数 10

分类专栏： IQA 文章标签：人工智能 IQA 图像质量评估 PYIQA

于 2024-03-01 17:27:31 首次发布

本文链接：https://blog.csdn.net/zzy_pphz/article/details/136401553

版权

IQA 专栏收录该内容

1 篇文章

订阅专栏

本文介绍了Python图像质量评估库PTIQA的安装和使用，包括LPIPS、NIQE和FID等指标计算方法，区分了FR-IQA（参考图像质量评估）和NR-IQA（无参考图像质量评估）。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

IQA表示图像质量评估，PTIQA是python中的IQA库，里面内置了多个图像质量评估的工具。其安装以及使用方法如下：

1.安装PYIQA

pip install pyiqa

使用下面语句，可以看到支持的IQA类别

2.使用方法如下：

lpips_metric = pyiqa.create_metric('lpips').cuda()
niqe_metric = pyiqa.create_metric('niqe').cuda()
fid_metric = pyiqa.create_metric('fid')

# 图片路径输入
lpips_score = lpips_metric('./ResultsCalibra/dist_dir/I03.bmp', './ResultsCalibra/ref_dir/I03.bmp')
niqe_score = niqe_metric('./ResultsCalibra/dist_dir/I03.bmp')

# tensor输入，tensor格式为RGB, [0, 1]
from pyiqa import imread2tensor
dist_img_tensor = imread2tensor('./ResultsCalibra/dist_dir/I03.bmp').unsqueeze(0).cuda()
ref_img_tensor = imread2tensor('./ResultsCalibra/ref_dir/I03.bmp').unsqueeze(0).cuda()
lpips_score = lpips_metric(dist_img_tensor, ref_img_tensor)
niqe_score = niqe_metric(dist_img_tensor, ref_img_tensor)

# 目录输入，仅适用于FID
fid_score = fid_metric('./ResultsCalibra/dist_dir/', './ResultsCalibra/ref_dir')

图像质量评估还分为有参考图像质量评估FR-IQA和无参考图像质量评估NR-IQA。

NR-IQA：