使用IQA-PyTorch库进行NIQE值计算

前言

最近在做超分的模型训练，需要一个评价指标，选用了NIQE值

NIQE

NIQE是一种无参考图像质量指标，数值越小代表图片质量越好

安装IQA-PyTorch

需求：

• Ubuntu >= 18.04
• Python >= 3.8
• Pytorch >= 1.8.1
• CUDA >= 10.1 (if use GPU)

pip install pyiqa

评估

实现过程非常简单

import pyiqa
import torch

niqe_metric = pyiqa.create_metric('niqe').cuda() #创建metric

niqe_score = niqe_metric('./results/input2_out.jpg') #评估

print(niqe_score)