pose_with_det_demo.py File Reference

Go to the source code of this file.

Namespaces
namespace	pose_with_det_demo

Variables
list	pose_with_det_demo.sys_path = ['/workspace/smlab', '/workspace']
str	pose_with_det_demo.det_cfg_path = '/workspace/smlab/smdetect/configs/yolo/yolov8.py'
str	pose_with_det_demo.pose_cfg_path = '/workspace/smlab/smpose/configs/mmpose/hrnet_trt.py'
str	pose_with_det_demo.image_path = '/media/safemotion/HDD5/pjm_test/test_sample/P001_220722/images/00054591.jpg'
str	pose_with_det_demo.save_path = '/workspace/smlab/smpose/demo/pose_with_det_demo.jpg'
	pose_with_det_demo.det_model = smrunner.build_model(det_cfg_path)
	pose_with_det_demo.pose_model = smrunner.build_model(pose_cfg_path)
	pose_with_det_demo.image = cv2.imread(image_path)
	pose_with_det_demo.det_result = det_model.run_detector(image)
	pose_with_det_demo.pose_result = pose_model.run_detector(image, det_result['det_bboxes'])
	pose_with_det_demo.vis_img = image.copy()