前言

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

NIQE

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

安装IQA-PyTorch

需求：

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

评估

实现过程非常简单

1
2
3
4
5
6
7
8
import pyiqa
import torch
 
niqe_metric = pyiqa.create_metric('niqe').cuda() #创建metric
 
niqe_score = niqe_metric('./results/input2_out.jpg') #评估
 
print(niqe_score)