from mmpose.apis import inference_bottom_up_pose_model, vis_pose_result def run_backbone_infer(img_ndarr, pose_model): # test a single image pose_results, _ = inference_bottom_up_pose_model(pose_model, img_ndarr) rst = vis_pose_result(pose_model, img_ndarr, pose_results, dataset='TopDownCocoWholeBodyDataset', thickness=2, radius=8, bbox_color='white') return pose_results, rst if __name__ == '__main__': pass