set up for new goals
This commit is contained in:
parent
74f1673cb3
commit
61a680e604
@ -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')):
|
||||
'''
|
||||
|
@ -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...)
|
||||
|
Loading…
Reference in New Issue
Block a user