From 29e27ce355f07d9644bff4072dfd01199d0f002d Mon Sep 17 00:00:00 2001 From: davidpagnon Date: Sat, 13 Jul 2024 22:05:03 +0200 Subject: [PATCH] minor fixes --- Pose2Sim/Pose2Sim.py | 2 +- Pose2Sim/personAssociation.py | 4 ++-- Pose2Sim/poseEstimation.py | 2 +- Pose2Sim/triangulation.py | 4 ++-- 4 files changed, 6 insertions(+), 6 deletions(-) diff --git a/Pose2Sim/Pose2Sim.py b/Pose2Sim/Pose2Sim.py index 3162386..bf16ada 100644 --- a/Pose2Sim/Pose2Sim.py +++ b/Pose2Sim/Pose2Sim.py @@ -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. ''' - 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 # from Pose2Sim.opensimProcessing import opensim_processing_all diff --git a/Pose2Sim/personAssociation.py b/Pose2Sim/personAssociation.py index 6c604a5..2829756 100644 --- a/Pose2Sim/personAssociation.py +++ b/Pose2Sim/personAssociation.py @@ -565,7 +565,7 @@ def recap_tracking(config_dict, error=0, nb_cams_excluded=0): # Read config_dict project_dir = config_dict.get('project').get('project_dir') # 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 session_dir = session_dir if 'Config.toml' in os.listdir(session_dir) else os.getcwd() multi_person = config_dict.get('project').get('multi_person') @@ -627,7 +627,7 @@ def track_2d_all(config_dict): # Read config_dict project_dir = config_dict.get('project').get('project_dir') # 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 session_dir = session_dir if 'Config.toml' in os.listdir(session_dir) else os.getcwd() multi_person = config_dict.get('project').get('multi_person') diff --git a/Pose2Sim/poseEstimation.py b/Pose2Sim/poseEstimation.py index e73b1ea..a72caca 100644 --- a/Pose2Sim/poseEstimation.py +++ b/Pose2Sim/poseEstimation.py @@ -326,7 +326,7 @@ def rtm_estimator(config_dict): # Read config project_dir = config_dict['project']['project_dir'] # 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 session_dir = session_dir if 'Config.toml' in os.listdir(session_dir) else os.getcwd() frame_range = config_dict.get('project').get('frame_range') diff --git a/Pose2Sim/triangulation.py b/Pose2Sim/triangulation.py index 4ed4b37..5ef2d73 100644 --- a/Pose2Sim/triangulation.py +++ b/Pose2Sim/triangulation.py @@ -330,7 +330,7 @@ def recap_triangulate(config_dict, error, nb_cams_excluded, keypoints_names, cam # Read config_dict project_dir = config_dict.get('project').get('project_dir') # 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 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] @@ -724,7 +724,7 @@ def triangulate_all(config_dict): # Read config_dict project_dir = config_dict.get('project').get('project_dir') # 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 session_dir = session_dir if 'Config.toml' in os.listdir(session_dir) else os.getcwd() multi_person = config_dict.get('project').get('multi_person')