pose_with_det_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_instance_segmentation, draw_bboxes_list
from smutils.utils_vis import vis_pose_coco_skeleton
from smutils.utils_image import crop_image
 
#경로 설정
det_cfg_path = '/workspace/smlab/smdetect/configs/yolo/yolov8.py' # yolov8
pose_cfg_path = '/workspace/smlab/smpose/configs/mmpose/hrnet_trt.py' # hrnet
image_path = '/media/safemotion/HDD5/pjm_test/test_sample/P001_220722/images/00054591.jpg'
save_path = '/workspace/smlab/smpose/demo/pose_with_det_demo.jpg'
 
# 모델 빌드
det_model = smrunner.build_model(det_cfg_path)
pose_model = smrunner.build_model(pose_cfg_path)
 
# 이미지 로드
image = cv2.imread(image_path)
 
# 모델 inference
det_result = det_model.run_detector(image)
pose_result = pose_model.run_detector(image, det_result['det_bboxes'])
 
# 시각화
vis_img = image.copy()
vis_img = draw_bboxes_list(vis_img, det_result['det_bboxes'], (255, 0, 0), 2, 'xyxy')
vis_img = vis_pose_coco_skeleton(vis_img, pose_result)
 
cv2.imwrite(save_path, vis_img)