[Feature] Support NIQE (#80)

* support niqe * add ut * fix typo * fix video niqe * fix comments * rename * fix comments * add docs * fix ut * update comments * update comments * update comments * update comments * update comments * add data to wheele package * update comments * update comments --------- Co-authored-by: C1rN09 <[email protected]>
open-mmlab · Mar 3, 2023 · a05c685 · a05c685
1 parent 2c60fa0
commit a05c685
Show file tree

Hide file tree

Showing 10 changed files with 573 additions and 3 deletions.
diff --git a/MANIFEST.in b/MANIFEST.in
@@ -0,0 +1 @@
+include mmeval/extra_data/niqe_pris_params.npz
diff --git a/docs/en/api/metrics.rst b/docs/en/api/metrics.rst
@@ -48,6 +48,7 @@ Metrics
    ConnectivityError
    DOTAMeanAP
    ROUGE
+   NaturalImageQualityEvaluator
    Perplexity
    KeypointEndPointError
    KeypointAUC

diff --git a/docs/zh_cn/api/metrics.rst b/docs/zh_cn/api/metrics.rst
@@ -48,6 +48,7 @@ Metrics
    ConnectivityError
    DOTAMeanAP
    ROUGE
+   NaturalImageQualityEvaluator
    Perplexity
    KeypointEndPointError
    KeypointAUC

diff --git a/mmeval/extra_data/niqe_pris_params.npz b/mmeval/extra_data/niqe_pris_params.npz
diff --git a/mmeval/metrics/__init__.py b/mmeval/metrics/__init__.py
@@ -20,6 +20,7 @@
 from .mean_iou import MeanIoU
 from .mse import MeanSquaredError
 from .multi_label import AveragePrecision, MultiLabelMetric
+from .niqe import NaturalImageQualityEvaluator
 from .oid_map import OIDMeanAP
 from .pck_accuracy import JhmdbPCKAccuracy, MpiiPCKAccuracy, PCKAccuracy
 from .perplexity import Perplexity
@@ -41,7 +42,7 @@
     'AveragePrecision', 'AVAMeanAP', 'BLEU', 'DOTAMeanAP',
     'SumAbsoluteDifferences', 'GradientError', 'MattingMeanSquaredError',
     'ConnectivityError', 'ROUGE', 'Perplexity', 'KeypointEndPointError',
-    'KeypointAUC', 'KeypointNME'
+    'KeypointAUC', 'KeypointNME', 'NaturalImageQualityEvaluator'
 ]
 
 _deprecated_msg = (