mmpose/apis/__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 .inference import (collect_multi_frames, inference_bottomup,
                        inference_topdown, init_model)
from .inference_3d import (collate_pose_sequence, convert_keypoint_definition,
                           extract_pose_sequence, inference_pose_lifter_model)
from .inference_tracking import _compute_iou, _track_by_iou, _track_by_oks
from .inferencers import MMPoseInferencer, Pose2DInferencer

__all__ = [
    'init_model', 'inference_topdown', 'inference_bottomup',
    'collect_multi_frames', 'Pose2DInferencer', 'MMPoseInferencer',
    '_track_by_iou', '_track_by_oks', '_compute_iou',
    'inference_pose_lifter_model', 'extract_pose_sequence',
    'convert_keypoint_definition', 'collate_pose_sequence',
]