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

comment unused functions

Browse Source
This commit is contained in:
davidpagnon 2023-12-17 20:08:34 +01:00
parent cc25269920
commit f446c3b61f
1 changed files with 79 additions and 79 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

158
Pose2Sim/Pose2Sim.py
View File

@ -214,37 +214,37 @@ def poseEstimation(config=None):
raise NotImplementedError('This has not been integrated yet. \nPlease read README.md for further explanation') raise NotImplementedError('This has not been integrated yet. \nPlease read README.md for further explanation')
# TODO # # TODO
# Determine the level at which the function is called (session:3, participant:2, trial:1) # # Determine the level at which the function is called (session:3, participant:2, trial:1)
level, config_dicts = read_config_files(config) # level, config_dicts = read_config_files(config)
if type(config)==dict: # if type(config)==dict:
config_dict = config_dicts[0] # config_dict = config_dicts[0]
if config_dict.get('project').get('project_dir') == None: # if config_dict.get('project').get('project_dir') == None:
raise ValueError('Please specify the project directory in config_dict:\n \ # raise ValueError('Please specify the project directory in config_dict:\n \
config_dict.get("project").update({"project_dir":"<YOUR_TRIAL_DIRECTORY>"})') # config_dict.get("project").update({"project_dir":"<YOUR_TRIAL_DIRECTORY>"})')
# Set up logging # # Set up logging
session_dir = os.path.realpath(os.path.join(config_dicts[0].get('project').get('project_dir'), '..', '..')) # session_dir = os.path.realpath(os.path.join(config_dicts[0].get('project').get('project_dir'), '..', '..'))
setup_logging(session_dir) # setup_logging(session_dir)
# Batch process all trials # # Batch process all trials
for config_dict in config_dicts: # for config_dict in config_dicts:
start = time.time() # start = time.time()
project_dir = os.path.realpath(config_dict.get('project').get('project_dir')) # project_dir = os.path.realpath(config_dict.get('project').get('project_dir'))
seq_name = os.path.basename(project_dir) # seq_name = os.path.basename(project_dir)
frame_range = config_dict.get('project').get('frame_range') # frame_range = config_dict.get('project').get('frame_range')
frames = ["all frames" if frame_range == [] else f"frames {frame_range[0]} to {frame_range[1]}"][0] # frames = ["all frames" if frame_range == [] else f"frames {frame_range[0]} to {frame_range[1]}"][0]
logging.info("\n\n---------------------------------------------------------------------") # logging.info("\n\n---------------------------------------------------------------------")
logging.info("Camera synchronization") # logging.info("Camera synchronization")
logging.info("---------------------------------------------------------------------") # logging.info("---------------------------------------------------------------------")
logging.info(f"\nProject directory: {project_dir}") # logging.info(f"\nProject directory: {project_dir}")
pose_estimation_all(config_dict) # pose_estimation_all(config_dict)
end = time.time() # end = time.time()
logging.info(f'Pose estimation took {end-start:.2f} s.') # logging.info(f'Pose estimation took {end-start:.2f} s.')
def synchronization(config=None): def synchronization(config=None):
@ -258,37 +258,37 @@ def synchronization(config=None):
raise NotImplementedError('This has not been integrated yet. \nPlease read README.md for further explanation') raise NotImplementedError('This has not been integrated yet. \nPlease read README.md for further explanation')
#TODO # #TODO
# Determine the level at which the function is called (session:3, participant:2, trial:1) # # Determine the level at which the function is called (session:3, participant:2, trial:1)
level, config_dicts = read_config_files(config) # level, config_dicts = read_config_files(config)
if type(config)==dict: # if type(config)==dict:
config_dict = config_dicts[0] # config_dict = config_dicts[0]
if config_dict.get('project').get('project_dir') == None: # if config_dict.get('project').get('project_dir') == None:
raise ValueError('Please specify the project directory in config_dict:\n \ # raise ValueError('Please specify the project directory in config_dict:\n \
config_dict.get("project").update({"project_dir":"<YOUR_TRIAL_DIRECTORY>"})') # config_dict.get("project").update({"project_dir":"<YOUR_TRIAL_DIRECTORY>"})')
# Set up logging # # Set up logging
session_dir = os.path.realpath(os.path.join(config_dicts[0].get('project').get('project_dir'), '..', '..')) # session_dir = os.path.realpath(os.path.join(config_dicts[0].get('project').get('project_dir'), '..', '..'))
setup_logging(session_dir) # setup_logging(session_dir)
# Batch process all trials # # Batch process all trials
for config_dict in config_dicts: # for config_dict in config_dicts:
start = time.time() # start = time.time()
project_dir = os.path.realpath(config_dict.get('project').get('project_dir')) # project_dir = os.path.realpath(config_dict.get('project').get('project_dir'))
seq_name = os.path.basename(project_dir) # seq_name = os.path.basename(project_dir)
frame_range = config_dict.get('project').get('frame_range') # frame_range = config_dict.get('project').get('frame_range')
frames = ["all frames" if frame_range == [] else f"frames {frame_range[0]} to {frame_range[1]}"][0] # frames = ["all frames" if frame_range == [] else f"frames {frame_range[0]} to {frame_range[1]}"][0]
logging.info("\n\n---------------------------------------------------------------------") # logging.info("\n\n---------------------------------------------------------------------")
logging.info("Camera synchronization") # logging.info("Camera synchronization")
logging.info("---------------------------------------------------------------------") # logging.info("---------------------------------------------------------------------")
logging.info(f"\nProject directory: {project_dir}") # logging.info(f"\nProject directory: {project_dir}")
synchronize_cams_all(config_dict) # synchronize_cams_all(config_dict)
end = time.time() # end = time.time()
logging.info(f'Synchronization took {end-start:.2f} s.') # logging.info(f'Synchronization took {end-start:.2f} s.')
def personAssociation(config=None): def personAssociation(config=None):
@ -433,40 +433,40 @@ def opensimProcessing(config=None):
raise NotImplementedError('This has not been integrated yet. \nPlease read README.md for further explanation') raise NotImplementedError('This has not been integrated yet. \nPlease read README.md for further explanation')
# TODO # # TODO
from Pose2Sim.opensimProcessing import opensim_processing_all # from Pose2Sim.opensimProcessing import opensim_processing_all
# Determine the level at which the function is called (session:3, participant:2, trial:1) # # Determine the level at which the function is called (session:3, participant:2, trial:1)
level, config_dicts = read_config_files(config) # level, config_dicts = read_config_files(config)
if type(config)==dict: # if type(config)==dict:
config_dict = config_dicts[0] # config_dict = config_dicts[0]
if config_dict.get('project').get('project_dir') == None: # if config_dict.get('project').get('project_dir') == None:
raise ValueError('Please specify the project directory in config_dict:\n \ # raise ValueError('Please specify the project directory in config_dict:\n \
config_dict.get("project").update({"project_dir":"<YOUR_TRIAL_DIRECTORY>"})') # config_dict.get("project").update({"project_dir":"<YOUR_TRIAL_DIRECTORY>"})')
# Set up logging # # Set up logging
session_dir = os.path.realpath(os.path.join(config_dicts[0].get('project').get('project_dir'), '..', '..')) # session_dir = os.path.realpath(os.path.join(config_dicts[0].get('project').get('project_dir'), '..', '..'))
setup_logging(session_dir) # setup_logging(session_dir)
# Batch process all trials # # Batch process all trials
for config_dict in config_dicts: # for config_dict in config_dicts:
start = time.time() # start = time.time()
project_dir = os.path.realpath(config_dict.get('project').get('project_dir')) # project_dir = os.path.realpath(config_dict.get('project').get('project_dir'))
seq_name = os.path.basename(project_dir) # seq_name = os.path.basename(project_dir)
frame_range = config_dict.get('project').get('frame_range') # frame_range = config_dict.get('project').get('frame_range')
frames = ["all frames" if frame_range == [] else f"frames {frame_range[0]} to {frame_range[1]}"][0] # frames = ["all frames" if frame_range == [] else f"frames {frame_range[0]} to {frame_range[1]}"][0]
logging.info("\n\n---------------------------------------------------------------------") # logging.info("\n\n---------------------------------------------------------------------")
# if static_file in project_dir: # # if static_file in project_dir:
# logging.info(f"Scaling model with <STATIC TRC FILE>.") # # logging.info(f"Scaling model with <STATIC TRC FILE>.")
# else: # # else:
# logging.info(f"Running inverse kinematics <MOTION TRC FILE>.") # # logging.info(f"Running inverse kinematics <MOTION TRC FILE>.")
logging.info("---------------------------------------------------------------------") # logging.info("---------------------------------------------------------------------")
logging.info(f"\nOpenSim output directory: {project_dir}") # logging.info(f"\nOpenSim output directory: {project_dir}")
opensim_processing_all(config_dict) # opensim_processing_all(config_dict)
end = time.time() # end = time.time()
logging.info(f'Model scaling took {end-start:.2f} s.') # logging.info(f'Model scaling took {end-start:.2f} s.')