nicetoolbox.detectors.method_detectors.body_joints.mmpose_inference.main

nicetoolbox.detectors.method_detectors.body_joints.mmpose_inference.main(config)[source]

Main function to run the MMPose inference.

Saves the results as npz files to the output folder with the following structure: - 2d: Numpy array of shape

(num_persons, num_cameras, num_frames, num_keypoints, [coordinate_x, coordinate_y, confidence_score])

  • bbox_2d: Numpy array of shape

    (num_persons, num_frames, 1, [top_left_x, top_left_y, bottom_right_x, bottom_right_y, confidence_score])

  • data_description: A dictionary containing the description of the data.

Parameters:

config (dict) – A dictionary containing the configuration parameters for the MMPose inference algorithm.