You can use any RTMlib model (hand, face, animal, or custom trained models) by providing a dictionary to the mode configuration. The dictionary must be enclosed in triple quotes and can contain local paths or URLs to .onnx or .zip files.
Key dictionary keys:
det_class: Detection class (e.g., 'YOLOX').det_model: Path/URL to the detection model.det_input_size: List of [width, height] for detection.pose_class: Pose class (e.g., 'RTMPose', 'RTMO').pose_model: Path/URL to the pose model.pose_input_size: List of [width, height] for pose estimation.
Example: Custom Balanced Mode
mode = """{'det_class':'YOLOX',
'det_model':'https://huggingface.co/datasets/DavidPagnon/rtmlib_models/resolve/main/mmpose/rtmposev1/onnx_sdk/yolox_m_8xb8-300e_humanart-c2c7a14a.onnx',
'det_input_size':[640, 640],
'pose_class':'RTMPose',
'pose_model':'https://huggingface.co/datasets/DavidPagnon/rtmlib_models/resolve/main/mmpose/rtmposev1/onnx_sdk/rtmpose-m_simcc-body7_pt-body7-halpe26_256x192-4d3e73dd_20230605.onnx',
'pose_input_size':[192,256]}"""
Example: One-stage RTMO model
mode = """{'pose_class':'RTMO',
'pose_model':'https://huggingface.co/datasets/DavidPagnon/rtmlib_models/resolve/main/mmpose/rtmo/onnx_sdk/rtmo-m_16xb8-600e_body7-640x640-39e78cc4_20231211.onnx',
'pose_input_size':[640, 640]}"""
# Equivalent to mode='balanced', with body_with_feet pose model
mode = """{'det_class':'YOLOX',
'det_model':'https://huggingface.co/datasets/DavidPagnon/rtmlib_models/resolve/main/mmpose/rtmposev1/onnx_sdk/yolox_m_8xb8-300e_humanart-c2c7a14a.onnx',
'det_input_size':[640, 640],
'pose_class':'RTMPose',
'pose_model':'https://huggingface.co/datasets/DavidPagnon/rtmlib_models/resolve/main/mmpose/rtmposev1/onnx_sdk/rtmpose-m_simcc-body7_pt-body7-halpe26_700e-256x192-4d3e73dd_20230605.onnx',
'pose_input_size':[192,256]}"""
# With one-stage RTMO model
mode = """{'pose_class':'RTMO',
'pose_model':'https://huggingface.co/datasets/DavidPagnon/rtmlib_models/resolve/main/mmpose/rtmo/onnx_sdk/rtmo-m_16xb16-600e_body7-640x640-39e78cc4_20231211.onnx',
'pose_input_size':[640, 640]}"""
# With animal pose estimation:
mode = """{'pose_class':'RTMPose',
'pose_model':'https://huggingface.co/datasets/DavidPagnon/rtmlib_models/resolve/main/mmpose/rtmposev1/onnx_sdk/rtmpose-m_simcc-ap10k_pt-aic-coco_210e-256x256-7a041aa1_20230206.onnx',
'pose_input_size':[256,256]}"""