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
|
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')):
|
def calibration(config=os.path.join('User', 'Config.toml')):
|
||||||
'''
|
'''
|
||||||
Cameras calibration from checkerboards or from qualisys files.
|
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')):
|
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')):
|
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>
|
</br>
|
||||||
|
|
||||||
> - [x] **Demo:** Provide Demo data for users to test the code.
|
> - [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.
|
> - [ ] **Tutorials:** Make video tutorials.
|
||||||
> - [ ] **Doc:** Use [Sphinx](https://www.sphinx-doc.org/en/master) or [MkDocs](https://www.mkdocs.org) for clearer documentation.
|
> - [ ] **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**
|
> - [x] **Pip package**
|
||||||
> - [ ] **Conda package**
|
> - [ ] **Conda package**
|
||||||
> - [ ] **Docker image**
|
> - [ ] **Docker image**
|
||||||
|
> - [ ] **** Run pose estimation and OpenSim from within Pose2Sim
|
||||||
> - [ ] **Run from command line via click or typer**
|
> - [ ] **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...)
|
> - [ ] **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