Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Feature] Support NIQE #80

Merged
merged 20 commits into from
Mar 3, 2023
Merged
1 change: 1 addition & 0 deletions MANIFEST.in
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
include mmeval/extra_data/niqe_pris_params.npz
1 change: 1 addition & 0 deletions docs/en/api/metrics.rst
Original file line number Diff line number Diff line change
Expand Up @@ -48,6 +48,7 @@ Metrics
ConnectivityError
DOTAMeanAP
ROUGE
NaturalImageQualityEvaluator
Perplexity
KeypointEndPointError
KeypointAUC
Expand Down
1 change: 1 addition & 0 deletions docs/zh_cn/api/metrics.rst
Original file line number Diff line number Diff line change
Expand Up @@ -48,6 +48,7 @@ Metrics
ConnectivityError
DOTAMeanAP
ROUGE
NaturalImageQualityEvaluator
Perplexity
KeypointEndPointError
KeypointAUC
Expand Down
Binary file added mmeval/extra_data/niqe_pris_params.npz
Binary file not shown.
3 changes: 2 additions & 1 deletion mmeval/metrics/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -41,7 +42,7 @@
'AveragePrecision', 'AVAMeanAP', 'BLEU', 'DOTAMeanAP',
'SumAbsoluteDifferences', 'GradientError', 'MattingMeanSquaredError',
'ConnectivityError', 'ROUGE', 'Perplexity', 'KeypointEndPointError',
'KeypointAUC', 'KeypointNME'
'KeypointAUC', 'KeypointNME', 'NaturalImageQualityEvaluator'
]

_deprecated_msg = (
Expand Down
Loading