pose_with_det_demo Namespace Reference

Variables
list	sys_path = ['/workspace/smlab', '/workspace']
str	det_cfg_path = '/workspace/smlab/smdetect/configs/yolo/yolov8.py'
str	pose_cfg_path = '/workspace/smlab/smpose/configs/mmpose/hrnet_trt.py'
str	image_path = '/media/safemotion/HDD5/pjm_test/test_sample/P001_220722/images/00054591.jpg'
str	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)
	det_result = det_model.run_detector(image)
	pose_result = pose_model.run_detector(image, det_result['det_bboxes'])
	vis_img = image.copy()

Variable Documentation

det_cfg_path

str pose_with_det_demo.det_cfg_path = '/workspace/smlab/smdetect/configs/yolo/yolov8.py'

Definition at line 16 of file pose_with_det_demo.py.

det_model

pose_with_det_demo.det_model = smrunner.build_model(det_cfg_path)

Definition at line 22 of file pose_with_det_demo.py.

det_result

pose_with_det_demo.det_result = det_model.run_detector(image)

Definition at line 29 of file pose_with_det_demo.py.

image

pose_with_det_demo.image = cv2.imread(image_path)

Definition at line 26 of file pose_with_det_demo.py.

image_path

str pose_with_det_demo.image_path = '/media/safemotion/HDD5/pjm_test/test_sample/P001_220722/images/00054591.jpg'

Definition at line 18 of file pose_with_det_demo.py.

pose_cfg_path

str pose_with_det_demo.pose_cfg_path = '/workspace/smlab/smpose/configs/mmpose/hrnet_trt.py'

Definition at line 17 of file pose_with_det_demo.py.

pose_model

pose_with_det_demo.pose_model = smrunner.build_model(pose_cfg_path)

Definition at line 23 of file pose_with_det_demo.py.

pose_result

pose_with_det_demo.pose_result = pose_model.run_detector(image, det_result['det_bboxes'])

Definition at line 30 of file pose_with_det_demo.py.

save_path

str pose_with_det_demo.save_path = '/workspace/smlab/smpose/demo/pose_with_det_demo.jpg'

Definition at line 19 of file pose_with_det_demo.py.

sys_path

list pose_with_det_demo.sys_path = ['/workspace/smlab', '/workspace']

Definition at line 3 of file pose_with_det_demo.py.

vis_img

pose_with_det_demo.vis_img = image.copy()

Definition at line 33 of file pose_with_det_demo.py.