pose_demo.py

Go to the documentation of this file.

#system path 설정
import sys
sys_path = ['/workspace/smlab', '/workspace']
for path in sys_path:
    if path not in sys.path:
        sys.path.append(path)
 
#import package
import cv2
import smrunner
from smutils.utils_vis import vis_pose_coco_skeleton
 
#경로 설정
cfg_path = '/workspace/smlab/smpose/configs/mmpose/hrnet_trt.py' # hrnet
image_path = '/workspace/smlab/smpose/demo/human-pose.jpg'
save_path = '/workspace/smlab/smpose/demo/skeleton.jpg'
 
# 모델 빌드
pose_model = smrunner.build_model(cfg_path)
 
# 이미지 로드
image = cv2.imread(image_path)
 
# 모델 inference
result = pose_model.run_detector(image)
 
#시각화 영상 만들기 및 저장
vis_img = image.copy()
vis_img = vis_pose_coco_skeleton(image, result)
cv2.imwrite(save_path, vis_img)
 
#모델 결과 출력
print(result)