mmpose/evaluation/metrics/__init__.py

# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.

from .coco_metric import CocoMetric
from .coco_wholebody_metric import CocoWholeBodyMetric
from .keypoint_2d_metrics import (AUC, EPE, NME, JhmdbPCKAccuracy,
                                  MpiiPCKAccuracy, PCKAccuracy)
from .keypoint_3d_metrics import MPJPE
from .keypoint_partition_metric import KeypointPartitionMetric
from .posetrack18_metric import PoseTrack18Metric
from .goliath_metric import GoliathMetric
from .goliath_coco_wholebody_metric import GoliathCocoWholeBodyMetric
from .goliath3d_coco_wholebody_metric import Goliath3dCocoWholeBodyMetric

__all__ = [
    'CocoMetric', 'PCKAccuracy', 'MpiiPCKAccuracy', 'JhmdbPCKAccuracy', 'AUC',
    'EPE', 'NME', 'PoseTrack18Metric', 'CocoWholeBodyMetric',
    'KeypointPartitionMetric', 'MPJPE', 'GoliathMetric',
    'GoliathCocoWholeBodyMetric', 'Goliath3dCocoWholeBodyMetric'
]