13 lines
454 B
Python
13 lines
454 B
Python
from mmpose.apis import inference_bottom_up_pose_model, vis_pose_result
|
|
import cv2
|
|
import os
|
|
|
|
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 |