mmpose/core/evaluation/__init__.py

# Copyright (c) OpenMMLab. All rights reserved.
from .bottom_up_eval import (aggregate_scale, aggregate_stage_flip,
                             flip_feature_maps, get_group_preds,
                             split_ae_outputs)
from .eval_hooks import DistEvalHook, EvalHook
from .mesh_eval import compute_similarity_transform
from .pose3d_eval import keypoint_3d_auc, keypoint_3d_pck, keypoint_mpjpe
from .top_down_eval import (keypoint_auc, keypoint_epe, keypoint_pck_accuracy,
                            keypoints_from_heatmaps, keypoints_from_heatmaps3d,
                            keypoints_from_regression,
                            multilabel_classification_accuracy,
                            pose_pck_accuracy, post_dark_udp)

__all__ = [
    'EvalHook', 'DistEvalHook', 'pose_pck_accuracy', 'keypoints_from_heatmaps',
    'keypoints_from_regression', 'keypoint_pck_accuracy', 'keypoint_3d_pck',
    'keypoint_3d_auc', 'keypoint_auc', 'keypoint_epe', 'get_group_preds',
    'split_ae_outputs', 'flip_feature_maps', 'aggregate_stage_flip',
    'aggregate_scale', 'compute_similarity_transform', 'post_dark_udp',
    'keypoint_mpjpe', 'keypoints_from_heatmaps3d',
    'multilabel_classification_accuracy'
]