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

erged augmented to triangulation

Browse Source
This commit is contained in:
Kim HunMin 2024-01-13 13:15:09 +09:00
parent a601c3eac2
commit 0dfc7ea705
3 changed files with 95 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
Pose2Sim/Pose2Sim.py
View File

@ -345,6 +345,8 @@ def triangulation(config=None):
'''
from Pose2Sim.triangulation import triangulate_all
from Pose2Sim.augmenter import augmentTRC
from Pose2Sim.filtering import filter_all
# Determine the level at which the function is called (session:3, participant:2, trial:1)
level, config_dicts = read_config_files(config)
@ -371,7 +373,9 @@ def triangulation(config=None):
logging.info(f"Triangulation of 2D points for {seq_name}, for {frames}.")
logging.info("---------------------------------------------------------------------")
logging.info(f"\nProject directory: {project_dir}")
if config_dict.get('pose').get('pose_model') == 'BODY_25_AUGMENTED':
triangulation_filt_AUGMENTED(config_dict)
else:
triangulate_all(config_dict)
end = time.time()
@ -551,6 +555,7 @@ def triangulation_filt_AUGMENTED(config=None):
def filtering(config=None):
'''
Filter trc 3D coordinates.

8
Pose2Sim/filtering.py
View File

@ -464,9 +464,17 @@ def filter_all(config):
# Trc paths
trc_f_in = f'{seq_name}_{f_range[0]}-{f_range[1]}.trc'
trc_f_in_LSTM = f'{seq_name}_{f_range[0]}-{f_range[1]}_LSTM.trc'
trc_f_out = f'{seq_name}_filt_{filter_type}_{f_range[0]}-{f_range[1]}.trc'
trc_f_out_LSTM = f'{seq_name}_filt_{filter_type}_{f_range[0]}-{f_range[1]}_LSTM.trc'
if os.path.exists(os.path.join(pose3d_dir, trc_f_in_LSTM)):
trc_path_in = os.path.join(pose3d_dir, trc_f_in_LSTM)
trc_path_out = os.path.join(pose3d_dir, trc_f_out_LSTM)
elif os.path.exists(os.path.join(pose3d_dir, trc_f_in)):
trc_path_in = os.path.join(pose3d_dir, trc_f_in)
trc_path_out = os.path.join(pose3d_dir, trc_f_out)
else:
raise FileNotFoundError("Not found .trc file.")
# Read trc header
with open(trc_path_in, 'r') as trc_file:

78
Pose2Sim/skeletons.py
View File

@ -77,6 +77,46 @@ BODY_25B = Node("CHip", id=None, children=[
]),
])
'''BODY_25B_AUGMENTED (full-body without hands, experimental, from OpenPose)
https://github.com/CMU-Perceptual-Computing-Lab/openpose_train/blob/master/experimental_models/README.md'''
BODY_25B_AUGMENTED = Node("CHip", id=None, children=[
Node("RHip", id=12, children=[
Node("RKnee", id=14, children=[
Node("RAnkle", id=16, children=[
Node("RBigToe", id=22, children=[
Node("RSmallToe", id=23),
]),
Node("RHeel", id=24),
]),
]),
]),
Node("LHip", id=11, children=[
Node("LKnee", id=13, children=[
Node("LAnkle", id=15, children=[
Node("LBigToe", id=19, children=[
Node("LSmallToe", id=20),
]),
Node("LHeel", id=21),
]),
]),
]),
Node("Neck", id=17, children=[
Node("Head", id=18, children=[
Node("Nose", id=0),
]),
Node("RShoulder", id=6, children=[
Node("RElbow", id=8, children=[
Node("RWrist", id=10),
]),
]),
Node("LShoulder", id=5, children=[
Node("LElbow", id=7, children=[
Node("LWrist", id=9),
]),
]),
]),
])
'''BODY_25 (full-body without hands, standard, from OpenPose)
https://github.com/CMU-Perceptual-Computing-Lab/openpose/tree/master/models'''
@ -116,6 +156,44 @@ BODY_25 = Node("CHip", id=8, children=[
]),
])
'''BODY_25_AUGMENTED (full-body without hands, standard, from OpenPose)
https://github.com/CMU-Perceptual-Computing-Lab/openpose/tree/master/models'''
BODY_25_AUGMENTED = Node("CHip", id=8, children=[
Node("RHip", id=9, children=[
Node("RKnee", id=10, children=[
Node("RAnkle", id=11, children=[
Node("RBigToe", id=22, children=[
Node("RSmallToe", id=23),
]),
Node("RHeel", id=24),
]),
]),
]),
Node("LHip", id=12, children=[
Node("LKnee", id=13, children=[
Node("LAnkle", id=14, children=[
Node("LBigToe", id=19, children=[
Node("LSmallToe", id=20),
]),
Node("LHeel", id=21),
]),
]),
]),
Node("Neck", id=17, children=[
Node("Nose", id=0),
Node("RShoulder", id=2, children=[
Node("RElbow", id=3, children=[
Node("RWrist", id=4),
]),
]),
Node("LShoulder", id=5, children=[
Node("LElbow", id=6, children=[
Node("LWrist", id=7),
]),
]),
]),
])
'''BODY_135 (full-body with hands and face, experimental, from OpenPose)
https://github.com/CMU-Perceptual-Computing-Lab/openpose_train/blob/master/experimental_models/README.md)'''