RTMO 姿态识别
预测示例:
# Copyright (c) OpenMMLab. All rights reserved.
from argparse import ArgumentParser
from typing import Dictfrom mmpose.apis.inferencers import MMPoseInferencer, get_model_aliasesfilter_args = dict(bbox_thr=0.3, nms_thr=0.3, pose_based_nms=False)
POSE2D_SPECIFIC_ARGS = dict(yoloxpose=dict(bbox_thr=0.01, nms_thr=0.65, pose_based_nms=True), rtmo=dict(bbox_thr=0.1, nms_thr=0.65, pose_based_nms=True), )def parse_args():parser = ArgumentParser()parser.add_argument('inputs', type=str, nargs='?',default=r'D:\data\tiaosheng\imgs', help='Input image/video path or folder path.')# init argsparser.add_argument('--pose2d', type=str, default='rtmo', help='Pretrained 2D pose estimation algorithm. It\'s the path to the '