Skip to content
New issue

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

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

Already on GitHub? # to your account

[Refactor] Refactor single and multi label metrics #81

Merged
merged 7 commits into from
Mar 9, 2023
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 13 additions & 9 deletions mmeval/metrics/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@

from .accuracy import Accuracy
from .ava_map import AVAMeanAP
from .average_precision import AveragePrecision
from .bleu import BLEU
from .coco_detection import COCODetection
from .connectivity_error import ConnectivityError
Expand All @@ -19,32 +20,33 @@
from .matting_mse import MattingMeanSquaredError
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
from .precision_recall_f1score import (MultiLabelPrecisionRecallF1score,
PrecisionRecallF1score,
SingleLabelPrecisionRecallF1score)
from .proposal_recall import ProposalRecall
from .psnr import PeakSignalNoiseRatio
from .rouge import ROUGE
from .sad import SumAbsoluteDifferences
from .single_label import SingleLabelMetric
from .snr import SignalNoiseRatio
from .ssim import StructuralSimilarity
from .voc_map import VOCMeanAP
from .word_accuracy import WordAccuracy

__all__ = [
'Accuracy', 'MeanIoU', 'VOCMeanAP', 'OIDMeanAP', 'EndPointError',
'F1Score', 'HmeanIoU', 'SingleLabelMetric', 'COCODetection', 'PCKAccuracy',
'MpiiPCKAccuracy', 'JhmdbPCKAccuracy', 'ProposalRecall',
'PeakSignalNoiseRatio', 'MeanAbsoluteError', 'MeanSquaredError',
'StructuralSimilarity', 'SignalNoiseRatio', 'MultiLabelMetric',
'AveragePrecision', 'AVAMeanAP', 'BLEU', 'DOTAMeanAP',
'F1Score', 'HmeanIoU', 'COCODetection', 'PCKAccuracy', 'MpiiPCKAccuracy',
'JhmdbPCKAccuracy', 'ProposalRecall', 'PeakSignalNoiseRatio',
'MeanAbsoluteError', 'MeanSquaredError', 'StructuralSimilarity',
'SignalNoiseRatio', 'AveragePrecision', 'AVAMeanAP', 'BLEU', 'DOTAMeanAP',
'SumAbsoluteDifferences', 'GradientError', 'MattingMeanSquaredError',
'ConnectivityError', 'ROUGE', 'Perplexity', 'KeypointEndPointError',
'KeypointAUC', 'KeypointNME', 'NaturalImageQualityEvaluator',
'WordAccuracy'
'WordAccuracy', 'PrecisionRecallF1score',
'SingleLabelPrecisionRecallF1score', 'MultiLabelPrecisionRecallF1score'
]

_deprecated_msg = (
Expand All @@ -61,7 +63,9 @@
'SNR': 'SignalNoiseRatio',
'SSIM': 'StructuralSimilarity',
'SAD': 'SumAbsoluteDifferences',
'MattingMSE': 'MattingMeanSquaredError'
'MattingMSE': 'MattingMeanSquaredError',
'SingleLabelMetric': 'SingleLabelPrecisionRecallF1score',
'MultiLabelMetric': 'MultiLabelPrecisionRecallF1score'
}


Expand Down
Loading