set up for new goals

This commit is contained in:
davidpagnon 2023-09-17 20:48:13 +02:00
parent 74f1673cb3
commit 61a680e604
2 changed files with 44 additions and 3 deletions

View File

@ -86,6 +86,28 @@ def base_params(config_dict):
return project_dir, seq_name, frames
def poseEstimation(config=os.path.join('User', 'Config.toml')):
'''
Estimate pose using BlazePose, OpenPose, AlphaPose, or DeepLabCut
'''
from Pose2Sim.poseEstimation import pose_estimation_all
config_dict = read_config_file(config)
project_dir, seq_name, frames = base_params(config_dict)
logging.info("\n\n---------------------------------------------------------------------")
logging.info("Pose estimation")
logging.info("---------------------------------------------------------------------")
logging.info(f"\nProject directory: {project_dir}")
start = time.time()
pose_estimation_all(config_dict)
end = time.time()
logging.info(f'Pose estimation took {end-start:.2f} s.')
def calibration(config=os.path.join('User', 'Config.toml')):
'''
Cameras calibration from checkerboards or from qualisys files.
@ -110,10 +132,25 @@ def calibration(config=os.path.join('User', 'Config.toml')):
def synchronization(config=os.path.join('User', 'Config.toml')):
'''
Synchronize if needed
Synchronize cameras if needed
'''
pass
from Pose2Sim.synchronization import synchronize_cams_all
config_dict = read_config_file(config)
project_dir, seq_name, frames = base_params(config_dict)
logging.info("\n\n---------------------------------------------------------------------")
logging.info("Camera synchronization")
logging.info("---------------------------------------------------------------------")
logging.info(f"\nProject directory: {project_dir}")
start = time.time()
synchronize_cams_all(config_dict)
end = time.time()
logging.info(f'Synchronization took {end-start:.2f} s.')
def personAssociation(config=os.path.join('User', 'Config.toml')):
'''

View File

@ -855,6 +855,9 @@ If you want to contribute to Pose2Sim, please follow [this guide](https://docs.g
</br>
> - [x] **Demo:** Provide Demo data for users to test the code.
> - [ ] **Demo:** Add videos for users to experiment with other pose detecction frameworks
> - [ ] **Demo:** Time shift videos and json to demonstrate synchronization
> - [ ] **Demo:** Add another virtual person to demonstrate personAssociation
> - [ ] **Tutorials:** Make video tutorials.
> - [ ] **Doc:** Use [Sphinx](https://www.sphinx-doc.org/en/master) or [MkDocs](https://www.mkdocs.org) for clearer documentation.
@ -864,5 +867,6 @@ If you want to contribute to Pose2Sim, please follow [this guide](https://docs.g
> - [x] **Pip package**
> - [ ] **Conda package**
> - [ ] **Docker image**
> - [ ] **** Run pose estimation and OpenSim from within Pose2Sim
> - [ ] **Run from command line via click or typer**
> - [ ] **Utilities**: Export other data from c3d files into .mot or .sto files (angles, powers, forces, moments, GRF, EMG...)