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

minor fixes

Browse Source
This commit is contained in:
davidpagnon 2024-07-13 22:05:03 +02:00
parent 05c4502735
commit 29e27ce355
4 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Pose2Sim/Pose2Sim.py
View File

@ -502,7 +502,7 @@ def opensimProcessing(config=None):
or the function can be called without an argument, in which case it the config directory is the current one. or the function can be called without an argument, in which case it the config directory is the current one.
''' '''
raise NotImplementedError('This has not been integrated yet. \nPlease read README.md for further explanation') raise NotImplementedError('This has not been implemented yet. \nPlease see README.md for further explanation')
# # TODO # # TODO
# from Pose2Sim.opensimProcessing import opensim_processing_all # from Pose2Sim.opensimProcessing import opensim_processing_all

4
Pose2Sim/personAssociation.py
View File

@ -565,7 +565,7 @@ def recap_tracking(config_dict, error=0, nb_cams_excluded=0):
# Read config_dict # Read config_dict
project_dir = config_dict.get('project').get('project_dir') project_dir = config_dict.get('project').get('project_dir')
# if batch # if batch
session_dir = os.path.realpath(os.path.join(project_dir, '..', '..')) session_dir = os.path.realpath(os.path.join(project_dir, '..'))
# if single trial # if single trial
session_dir = session_dir if 'Config.toml' in os.listdir(session_dir) else os.getcwd() session_dir = session_dir if 'Config.toml' in os.listdir(session_dir) else os.getcwd()
multi_person = config_dict.get('project').get('multi_person') multi_person = config_dict.get('project').get('multi_person')
@ -627,7 +627,7 @@ def track_2d_all(config_dict):
# Read config_dict # Read config_dict
project_dir = config_dict.get('project').get('project_dir') project_dir = config_dict.get('project').get('project_dir')
# if batch # if batch
session_dir = os.path.realpath(os.path.join(project_dir, '..', '..')) session_dir = os.path.realpath(os.path.join(project_dir, '..'))
# if single trial # if single trial
session_dir = session_dir if 'Config.toml' in os.listdir(session_dir) else os.getcwd() session_dir = session_dir if 'Config.toml' in os.listdir(session_dir) else os.getcwd()
multi_person = config_dict.get('project').get('multi_person') multi_person = config_dict.get('project').get('multi_person')

2
Pose2Sim/poseEstimation.py
View File

@ -326,7 +326,7 @@ def rtm_estimator(config_dict):
# Read config # Read config
project_dir = config_dict['project']['project_dir'] project_dir = config_dict['project']['project_dir']
# if batch # if batch
session_dir = os.path.realpath(os.path.join(project_dir, '..', '..')) session_dir = os.path.realpath(os.path.join(project_dir, '..'))
# if single trial # if single trial
session_dir = session_dir if 'Config.toml' in os.listdir(session_dir) else os.getcwd() session_dir = session_dir if 'Config.toml' in os.listdir(session_dir) else os.getcwd()
frame_range = config_dict.get('project').get('frame_range') frame_range = config_dict.get('project').get('frame_range')

4
Pose2Sim/triangulation.py
View File

@ -330,7 +330,7 @@ def recap_triangulate(config_dict, error, nb_cams_excluded, keypoints_names, cam
# Read config_dict # Read config_dict
project_dir = config_dict.get('project').get('project_dir') project_dir = config_dict.get('project').get('project_dir')
# if batch # if batch
session_dir = os.path.realpath(os.path.join(project_dir, '..', '..')) session_dir = os.path.realpath(os.path.join(project_dir, '..'))
# if single trial # if single trial
session_dir = session_dir if 'Config.toml' in os.listdir(session_dir) else os.getcwd() session_dir = session_dir if 'Config.toml' in os.listdir(session_dir) else os.getcwd()
calib_dir = [os.path.join(session_dir, c) for c in os.listdir(session_dir) if os.path.isdir(os.path.join(session_dir, c)) and 'calib' in c.lower()][0] calib_dir = [os.path.join(session_dir, c) for c in os.listdir(session_dir) if os.path.isdir(os.path.join(session_dir, c)) and 'calib' in c.lower()][0]
@ -724,7 +724,7 @@ def triangulate_all(config_dict):
# Read config_dict # Read config_dict
project_dir = config_dict.get('project').get('project_dir') project_dir = config_dict.get('project').get('project_dir')
# if batch # if batch
session_dir = os.path.realpath(os.path.join(project_dir, '..', '..')) session_dir = os.path.realpath(os.path.join(project_dir, '..'))
# if single trial # if single trial
session_dir = session_dir if 'Config.toml' in os.listdir(session_dir) else os.getcwd() session_dir = session_dir if 'Config.toml' in os.listdir(session_dir) else os.getcwd()
multi_person = config_dict.get('project').get('multi_person') multi_person = config_dict.get('project').get('multi_person')