fulyvv/pose2sim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

change openpose_model to pose_model

Browse Source
This commit is contained in:
davidpagnon 2022-09-29 18:07:53 +02:00
parent 7d8658e7ce
commit 6d54409bc2
5 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Pose2Sim/Demo/User/Config.toml
View File

@ -19,7 +19,7 @@ pose3d_folder_name = 'pose-3d'
opensim_folder_name = 'opensim'
[pose-2d]
openpose_model = 'BODY_25B' #BODY_25B, BODY_25, COCO, MPI are available.
pose_model = 'BODY_25B' #BODY_25B, BODY_25, COCO, MPI are available.
# set your own model in skeleton.py if you don't use openpose.
[calibration]

2
Pose2Sim/Demo/User/test.toml
View File

@ -19,7 +19,7 @@ pose3d_folder_name = 'pose-3d'
opensim_folder_name = 'opensim'
[pose-2d]
openpose_model = 'BODY_25B' #BODY_25B, BODY_25, COCO, MPI are available.
pose_model = 'BODY_25B' #BODY_25B, BODY_25, COCO, MPI are available.
# set your own model in skeleton.py if you don't use openpose.
[calibration]

2
Pose2Sim/Empty_project/User/Config.toml
View File

@ -19,7 +19,7 @@ pose3d_folder_name = 'pose-3d'
opensim_folder_name = 'opensim'
[pose-2d]
openpose_model = 'BODY_25B' #BODY_25B, BODY_25, COCO, MPI are available.
pose_model = 'BODY_25B' #BODY_25B, BODY_25, COCO, MPI are available.
# set your own model in skeleton.py if you don't use openpose.
[calibration]

4
Pose2Sim/track_2d.py
View File

@ -244,7 +244,7 @@ def track_2d_all(config):
calib_folder_name = config.get('project').get('calib_folder_name')
poseTracked_folder_name = config.get('project').get('poseTracked_folder_name')
pose_folder_name = config.get('project').get('pose_folder_name')
openpose_model = config.get('pose-2d').get('openpose_model')
pose_model = config.get('pose-2d').get('pose_model')
tracked_keypoint = config.get('2d-tracking').get('tracked_keypoint')
json_folder_extension = config.get('project').get('pose_json_folder_extension')
frames_range = config.get('project').get('frames_range')
@ -258,7 +258,7 @@ def track_2d_all(config):
P = computeP(calib_file)
# selection of tracked keypoint id
model = eval(openpose_model)
model = eval(pose_model)
tracked_keypoint_id = [node.id for _, _, node in RenderTree(model) if node.name==tracked_keypoint][0]
# 2d-pose files selection

4
Pose2Sim/triangulate_3d.py
View File

@ -359,7 +359,7 @@ def triangulate_all(config):
project_dir = config.get('project').get('project_dir')
if project_dir == '': project_dir = os.getcwd()
calib_folder_name = config.get('project').get('calib_folder_name')
openpose_model = config.get('pose-2d').get('openpose_model')
pose_model = config.get('pose-2d').get('pose_model')
pose_folder_name = config.get('project').get('pose_folder_name')
json_folder_extension = config.get('project').get('pose_json_folder_extension')
frames_range = config.get('project').get('frames_range')
@ -375,7 +375,7 @@ def triangulate_all(config):
P = computeP(calib_file)
# Retrieve keypoints from model
model = eval(openpose_model)
model = eval(pose_model)
keypoints_ids = [node.id for _, _, node in RenderTree(model) if node.id!=None]
keypoints_names = [node.name for _, _, node in RenderTree(model) if node.id!=None]
keypoints_idx = list(range(len(keypoints_ids)))