Perform a markerless kinematic analysis from multiple calibrated views as a unified workflow from an OpenPose input to an OpenSim result.
Project description
Pose2Sim
N.B:. Please set undistort_points and handle_LR_swap to false for now since it currently leads to inaccuracies. I'll try to fix it soon.
News: Version 0.10.0:
OpenSim scaling and inverse kinematics are now integrated in Pose2Sim! No static trial needed.
Other recently added features: Pose estimation, Automatic camera synchronization, Multi-person analysis, Blender visualization, Marker augmentation, Batch processing.
To upgrade, type
pip install pose2sim --upgrade
Pose2Sim
provides a workflow for 3D markerless kinematics, as an alternative to traditional marker-based MoCap methods.
Pose2Sim is free and open-source, low-cost but with research-grade accuracy and production-grade robustness. It gives a maximum of control on clearly explained parameters. Any combination of phones, webcams, or GoPros can be used with fully clothed subjects, so it is particularly adapted to the sports field, the doctor's office, or for outdoor 3D animation capture.
Note: For real-time analysis with a single camera, please consider Sports2D (note that the motion must lie in the sagittal or frontal plane).
Fun fact:
Pose2Sim stands for "OpenPose to OpenSim", as it originally used OpenPose inputs (2D keypoints coordinates) and lead to an OpenSim result (full-body 3D joint angles). Pose estimation is now performed with more recent models from RTMPose, and custom models (from DeepLabCut for example) can also be used.
Pose2Sim releases:
- v0.1 (08/2021): Published paper
- v0.2 (01/2022): Published code
- v0.3 (01/2023): Supported other pose estimation algorithms
- v0.4 (07/2023): New calibration tool based on scene measurements
- v0.5 (12/2023): Automatic batch processing
- v0.6 (02/2024): Marker augmentation, Blender visualizer
- v0.7 (03/2024): Multi-person analysis
- v0.8 (04/2024): New synchronization tool
- v0.9 (07/2024): Integration of pose estimation in the pipeline
- v0.10 (09/2024): Integration of OpenSim in the pipeline
- v0.11: Migrated documentation to new github.io website
- v0.12: Calibration based on keypoint detection, Handling left/right swaps, Correcting lens distortions
- v0.13: Graphical User Interface
- v1.0: First full release
N.B.: As always, I am more than happy to welcome contributors (see How to contribute).
Contents
- Installation and Demonstration
- Use on your own data
- Utilities
- How to cite and how to contribute
Installation and Demonstration
Installation
-
Optional:
Install Anaconda or Miniconda for simplicity and avoiding the risk of incompatibilities between libraries.Once installed, open an Anaconda prompt and create a virtual environment:
conda create -n Pose2Sim python=3.9 -y conda activate Pose2Sim
-
Install OpenSim:
Install the OpenSim Python API (if you do not want to install via conda, refer to this page):conda install -c opensim-org opensim -y
-
Install Pose2Sim:
If you don't use Anaconda, typepython -V
in terminal to make sure python>=3.9 is installed.-
OPTION 1: Quick install: Open a terminal.
pip install pose2sim
-
OPTION 2: Build from source and test the last changes: Open a terminal in the directory of your choice and Clone the Pose2Sim repository.
git clone --depth 1 https://github.com/perfanalytics/pose2sim.git cd pose2sim pip install .
-
-
Optional:
For faster inference, you can run on the GPU. Install pyTorch with CUDA and cuDNN support, and ONNX Runtime with GPU support (not available on MacOS).
Be aware that GPU support takes an additional 6 GB on disk. The full installation is then 10.75 GB instead of 4.75 GB.Run
nvidia-smi
in a terminal. If this results in an error, your GPU is probably not compatible with CUDA. If not, note the "CUDA version": it is the latest version your driver is compatible with (more information on this post).Then go to the ONNXruntime requirement page, note the latest compatible CUDA and cuDNN requirements. Next, go to the pyTorch website and install the latest version that satisfies these requirements (beware that torch 2.4 ships with cuDNN 9, while torch 2.3 installs cuDNN 8). For example:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
Finally, install ONNX Runtime with GPU support:
pip install onnxruntime-gpu
Check that everything went well within Python with these commands:
import torch; import onnxruntime as ort print(torch.cuda.is_available(), ort.get_available_providers()) # Should print "True ['CUDAExecutionProvider', ...]"
Note on storage use:
A full installation takes up to 11 GB of storage spate. However, GPU support is not mandatory and takes about 6 GB. Moreover, marker augmentation requires Tensorflow and does not necessarily yield better results. You can save an additional 1.3 GB by uninstalling it:pip uninstall tensorflow
.
A minimal installation with carefully chosen pose models and without GPU support, Tensorflow, PyQt5 would take less than 3 GB.
Demonstration Part-1: End to end video to 3D joint angle computation
This demonstration provides an example experiment of a person balancing on a beam, filmed with 4 cameras.
Open a terminal, enter pip show pose2sim
, report package location.
Copy this path and go to the Single participant Demo folder: cd <path>\Pose2Sim\Demo_SinglePerson
.
Type ipython
, and try the following code:
from Pose2Sim import Pose2Sim
Pose2Sim.calibration()
Pose2Sim.poseEstimation()
Pose2Sim.synchronization()
Pose2Sim.personAssociation()
Pose2Sim.triangulation()
Pose2Sim.filtering()
Pose2Sim.markerAugmentation()
Pose2Sim.kinematics()
3D marker locations are stored as .trc files in each trial folder in the pose-3d
directory.
3D joint angles are stored as .mot files in the kinematics
directory. Scaled models are also stored in the same directory.
Note:
- Default parameters have been provided in Config.toml but can be edited.
All of them are clearly documented: feel free to play with them! - You can run all stages at once:
from Pose2Sim import Pose2Sim Pose2Sim.runAll() # or: Pose2Sim.runAll(do_calibration=True, do_poseEstimation=True, do_synchronization=True, do_personAssociation=True, do_triangulation=True, do_filtering=True, do_markerAugmentation=True, do_kinematics=True)
- Try the calibration tool by changing
calibration_type
tocalculate
instead ofconvert
in Config.toml (more info there). - Note that Pose2Sim.markerAugmentation() does not necessarily improve results--in fact, results are worse half of the time. You can choose to not run this command, and save an additional 1.3 GB by uninstalling tensorflow:
pip uninstall tensorflow
.
Demonstration Part-2: Visualize your results with OpenSim or Blender
Visualize your results and look in detail for potential areas of improvement.
Basic visualization with the OpenSim GUI
-
Install OpenSim GUI:
Download the executable there. -
Visualize results:
- Open the OpenSim GUI, go to File > Open Model, and select the scaled model in the
kinematics
folder. - Go to File > Load Motion, and load the joint angle .mot file in the
kinematics
folder. - If you want to see the 3D marker locations, go to File > Preview Experimental Data, and load the .trc file in the
pose-3d
folder.
- Open the OpenSim GUI, go to File > Open Model, and select the scaled model in the
Further check with the Pose2Sim Blender add-on
-
Install the add-on:
Follow instructions on the Pose2Sim_Blender add-on page. -
Visualize results:
Just play with the buttons!
Visualize camera positions, videos, triangulated keypoints, OpenSim skeleton, video overlay your results on videos, ... or let your creativity flow and create your own animations!https://github.com/davidpagnon/Pose2Sim_Blender/assets/54667644/a2cfb75d-a2d4-471a-b6f8-8f1ee999a619
N.B.: Full install only required to import the skeleton. See instructions there.
Demonstration Part-3: Try multi-person analysis
Another person, hidden all along, will appear when multi-person analysis is activated!
Go to the Multi-participant Demo folder: cd <path>\Pose2Sim\Demo_MultiPerson
.
Type ipython
, and try the following code:
from Pose2Sim import Pose2Sim
Pose2Sim.calibration()
Pose2Sim.poseEstimation()
# Pose2Sim.synchronization()
Pose2Sim.personAssociation()
Pose2Sim.triangulation()
Pose2Sim.filtering()
Pose2Sim.markerAugmentation()
Pose2Sim.kinematics()
or equivalently:
from Pose2Sim import Pose2Sim
Pose2Sim.runAll(do_synchronization=False) # Synchronization possible, but tricky with multiple persons
One .trc file per participant will be generated and stored in the pose-3d
directory.
Similarly, one scaled .osim model and one joint angle .mot file per participant will be stored in the kinematics
folder.
You can visualize your results with Blender as explained in Demonstration Part-2.
N.B.:
- In Config.toml, set
project
>multi_person = true
for each trial that contains multiple persons. - Make sure that the order of
markerAugmentation
>participant_height
andparticipant_mass
matches the person's IDs.
Demonstration Part-4: Try batch processing
Run numerous analysis with different parameters and minimal friction.
Go to the Batch Demo folder: cd <path>\Pose2Sim\Demo_Batch
.
Type ipython
, and try the following code:
from Pose2Sim import Pose2Sim
Pose2Sim.calibration()
Pose2Sim.poseEstimation()
Pose2Sim.synchronization()
Pose2Sim.personAssociation()
Pose2Sim.triangulation()
Pose2Sim.filtering()
Pose2Sim.markerAugmentation()
Pose2Sim.kinematics()
or equivalently:
from Pose2Sim import Pose2Sim
Pose2Sim.runAll()
The batch processing structure requires a Config.toml
file in each of the trial directories. Global parameters are given in the Config.toml
file of the BatchSession
folder. They can be altered for specific Trials by uncommenting keys and their values in their respective Config.toml
files.
Run Pose2Sim from the BatchSession
folder if you want to batch process the whole session, or from a Trial
folder if you want to process only a specific trial.
SingleTrial | BatchSession |
---|---|
SingleTrial |
BatchSession |
For example, try uncommenting [project]
and set frame_range = [10,99]
, or uncomment [pose]
and set mode = 'lightweight'
in the Config.toml
file of Trial_2
.
Use on your own data
N.B.: If a step is not relevant for your use case (synchronization, person association, marker augmentation...), you can just skip it.
Setting up your project
Get yourself comfy!
- Open a terminal, enter
pip show pose2sim
, report package location.
Copy this path and docd <path>\pose2sim
. - Copy-paste the Demo_SinglePerson, Demo_MultiPerson, or Demo_Batch folder wherever you like, and rename it as you wish.
- The rest of the tutorial will explain to you how to populate the
Calibration
andvideos
folders, edit the Config.toml files, and run each Pose2Sim step.
2D pose estimation
Estimate 2D pose from images with RTMPose or another pose estimation solution.
With RTMPose (default):
RTMPose is a state-of-the-art pose estimation solution that is faster and more accurate than OpenPose. It is now included in Pose2Sim for straightforward end-to-end analysis.
Open an Anaconda prompt or a terminal in a Session
or Trial
folder.
Type ipython
.
from Pose2Sim import Pose2Sim
Pose2Sim.poseEstimation()
N.B.: Pose estimation can be run in lightweight
, balanced
, or performance
mode.
N.B.: The pose_model
with body, feet, hands, and face is required for wrist motion but is much slower and slightly less accurate on body keypoints.
N.B.: The GPU
will be used with ONNX backend if a valid CUDA installation is found (or MPS with MacOS), otherwise the CPU
will be used with OpenVINO backend.
N.B.: Pose estimation can be dramatically sped up by increasing the value of det_frequency
. In that case, the detection is only done every det_frequency
frames, and bounding boxes are tracked inbetween (keypoint detection is still performed on all frames).
With MMPose (coming soon):
Coming soon
With DeepLabCut:
If you need to detect specific points on a human being, an animal, or an object, you can also train your own model with DeepLabCut. In this case, Pose2Sim is used as an alternative to AniPose.
- Train your DeepLabCut model and run it on your images or videos (more instruction on their repository)
- Translate the h5 2D coordinates to json files (with
DLC_to_OpenPose.py
script, see Utilities). Note that the names of your camera folders must follow the same order as in the calibration file, and end with '_json':python -m DLC_to_OpenPose -i input_h5_file
- Edit
pose.CUSTOM
in Config.toml, and edit the node IDs so that they correspond to the column numbers of the 2D pose file, starting from zero. Make sure you also changed thepose_model
and thetracked_keypoint
.
You can visualize your skeleton's hierarchy by changing pose_model to CUSTOM and writing these lines:config_path = r'path_to_Config.toml' import toml, anytree config = toml.load(config_path) pose_model = config.get('pose').get('pose_model') model = anytree.importer.DictImporter().import_(config.get('pose').get(pose_model)) for pre, _, node in anytree.RenderTree(model): print(f'{pre}{node.name} id={node.id}')
- Create an OpenSim model if you need inverse kinematics.
With OpenPose (legacy):
N.B.: RTMlib is faster, more accurate, and easier to install than OpenPose. This is a legacy option.
N.B.: OpenPose model files are apparently not available on their website anymore. Send me an email at dp2032@bath.ac.uk if you want me to forward them to you!
The accuracy and robustness of Pose2Sim have been thoroughly assessed only with OpenPose, BODY_25B model. Consequently, we recommend using this 2D pose estimation solution. See OpenPose repository for installation and running. Windows portable demo is enough.
- Open a command prompt in your OpenPose directory.
Launch OpenPose for eachvideos
folder:bin\OpenPoseDemo.exe --model_pose BODY_25B --video <PATH_TO_TRIAL_DIR>\videos\cam01.mp4 --write_json <PATH_TO_TRIAL_DIR>\pose\pose_cam01_json
- The BODY_25B model has more accurate results than the standard BODY_25 one and has been extensively tested for Pose2Sim.
You can also use the BODY_135 model, which allows for the evaluation of pronation/supination, wrist flexion, and wrist deviation.
All other OpenPose models (BODY_25, COCO, MPII) are also supported.
Make sure you modify the Config.toml file accordingly. - Use one of the
json_display_with_img.py
orjson_display_with_img.py
scripts (see Utilities) if you want to display 2D pose detections.
With MediaPipe BlazePose (legacy):
N.B.: RTMlib is faster, more accurate, and easier to install than BlazePose. This is also a legacy option.
Mediapipe BlazePose is very fast, fully runs under Python, handles upside-down postures and wrist movements (but no subtalar ankle angles).
However, it is less robust and accurate than OpenPose, and can only detect a single person.
- Use the script
Blazepose_runsave.py
(see Utilities) to run BlazePose under Python, and store the detected coordinates in OpenPose (json) or DeepLabCut (h5 or csv) format:python -m Blazepose_runsave -i input_file -dJs
Type inpython -m Blazepose_runsave -h
for explanation on parameters. - Make sure you changed the
pose_model
and thetracked_keypoint
in the Config.toml file.
With AlphaPose (legacy):
N.B.: RTMlib is faster, more accurate, and easier to install than AlphaPose. This is also a legacy option.
AlphaPose is one of the main competitors of OpenPose, and its accuracy is comparable. As a top-down approach (unlike OpenPose which is bottom-up), it is faster on single-person detection, but slower on multi-person detection.
All AlphaPose models are supported (HALPE_26, HALPE_68, HALPE_136, COCO_133, COCO, MPII). For COCO and MPII, AlphaPose must be run with the flag "--format cmu".
- Install and run AlphaPose on your videos (more instruction on their repository)
- Translate the AlphaPose single json file to OpenPose frame-by-frame files (with
AlphaPose_to_OpenPose.py
script, see Utilities):python -m AlphaPose_to_OpenPose -i input_alphapose_json_file
- Make sure you changed the
pose_model
and thetracked_keypoint
in the Config.toml file.
Camera calibration
Calculate camera intrinsic properties and extrinsic locations and positions.
Convert a preexisting calibration file, or calculate intrinsic and extrinsic parameters from scratch.
Open an Anaconda prompt or a terminal in a Session
or Trial
folder.
Type ipython
.
from Pose2Sim import Pose2Sim
Pose2Sim.calibration()
Output file:
Convert from Caliscope, AniPose, FreeMocap, Qualisys, Optitrack, Vicon, OpenCap, EasyMocap, or bioCV
If you already have a calibration file, set calibration_type
type to convert
in your Config.toml file.
- From Caliscope, AniPose or FreeMocap:
- Copy your
.toml
calibration file to the Pose2SimCalibration
folder. - Calibration can be skipped since these formats are natively supported by Pose2Sim.
- Copy your
- From Qualisys:
- Export calibration to
.qca.txt
within QTM. - Copy it in the
Calibration
Pose2Sim folder. - set
convert_from
to 'qualisys' in your Config.toml file. Changebinning_factor
to 2 if you film in 540p.
- Export calibration to
- From Optitrack: Exporting calibration will be available in Motive 3.2. In the meantime:
- Calculate intrinsics with a board (see next section).
- Use their C++ API to retrieve extrinsic properties. Translation can be copied as is in your
Calib.toml
file, but TT_CameraOrientationMatrix first needs to be converted to a Rodrigues vector with OpenCV. See instructions here. - Use the
Calib.toml
file as is and do not run Pose2Sim.calibration()
- From Vicon:
- Copy your
.xcp
Vicon calibration file to the Pose2SimCalibration
folder. - set
convert_from
to 'vicon' in your Config.toml file. No other setting is needed.
- Copy your
- From OpenCap:
- Copy your
.pickle
OpenCap calibration files to the Pose2SimCalibration
folder. - set
convert_from
to 'opencap' in your Config.toml file. No other setting is needed.
- Copy your
- From EasyMocap:
- Copy your
intri.yml
andextri.yml
files to the Pose2SimCalibration
folder. - set
convert_from
to 'easymocap' in your Config.toml file. No other setting is needed.
- Copy your
- From bioCV:
- Copy your bioCV calibration files (no extension) to the Pose2Sim
Calibration
folder. - set
convert_from
to 'biocv' in your Config.toml file. No other setting is needed.
- Copy your bioCV calibration files (no extension) to the Pose2Sim
Calculate from scratch
Calculate calibration parameters with a checkerboard, with measurements on the scene, or automatically with detected keypoints.
Take heart, it is not that complicated once you get the hang of it!
N.B.: Try the calibration tool on the Demo by changing
calibration_type
tocalculate
in Config.toml.
For the sake of practicality, there are voluntarily few board images for intrinsic calibration, and few points to click for extrinsic calibration. In spite of this, your reprojection error should be under 1-2 cm, which does not hinder the quality of kinematic results in practice.).
-
Calculate intrinsic parameters with a checkerboard:
N.B.: Intrinsic parameters: camera properties (focal length, optical center, distortion), usually need to be calculated only once in their lifetime. In theory, cameras with same model and same settings will have identical intrinsic parameters.
N.B.: If you already calculated intrinsic parameters earlier, you can skip this step by settingoverwrite_intrinsics
to false.- Create a folder for each camera in your
Calibration\intrinsics
folder. - For each camera, film a checkerboard or a charucoboard. Either the board or the camera can be moved.
- Adjust parameters in the Config.toml file.
- Make sure that the board:
- is filmed from different angles, covers a large part of the video frame, and is in focus.
- is flat, without reflections, surrounded by a wide white border, and is not rotationally invariant (Nrows ≠ Ncols, and Nrows odd if Ncols even). Go to calib.io to generate a suitable checkerboard.
- A common error is to specify the external, instead of the internal number of corners (one less than the count from calib.io). This may be one less than you would intuitively think.
Intrinsic calibration error should be below 0.5 px.
- Create a folder for each camera in your
-
Calculate extrinsic parameters:
N.B.: Extrinsic parameters: camera placement in space (position and orientation), need to be calculated every time a camera is moved. Can be calculated from a board, or from points in the scene with known coordinates.
N.B.: If there is no measurable item in the scene, you can temporarily bring something in (a table, for example), perform calibration, and then remove it before you start capturing motion.- Create a folder for each camera in your
Calibration\extrinsics
folder. - Once your cameras are in place, shortly film either a board laid on the floor, or the raw scene
(only one frame is needed, but do not just take a photo unless you are sure it does not change the image format). - Adjust parameters in the Config.toml file.
- Then,
- With a checkerboard:
Make sure that it is seen by all cameras.
It should preferably be larger than the one used for intrinsics, as results will not be very accurate out of the covered zone. - With scene measurements (more flexible and potentially more accurate if points are spread out):
Manually measure the 3D coordinates of 10 or more points in the scene (tiles, lines on wall, boxes, treadmill dimensions...). These points should be as spread out as possible. Replaceobject_coords_3d
by these coordinates in Config.toml.
Then you will click on the corresponding image points for each view. - With keypoints:
For a more automatic calibration, OpenPose keypoints could also be used for calibration.
COMING SOON!
- With a checkerboard:
Extrinsic calibration error should be below 1 cm, but depending on your application, results will still be potentially acceptable up to 2.5 cm.
- Create a folder for each camera in your
Synchronizing, Associating, Triangulating, Filtering
Synchronization
2D points can be triangulated only if they represent the same body position across all cameras: therefore, views need to be synchronized.
For each camera, this computes mean vertical speed for the chosen keypoints, and finds the time offset for which their correlation is highest.
N.B.: Skip this step if your cameras are natively synchronized.
Open an Anaconda prompt or a terminal in a Session
or Trial
folder.
Type ipython
.
from Pose2Sim import Pose2Sim
Pose2Sim.synchronization()
In multi_person
mode, a video will pop up to let the user choose on which person to synchronize.
All keypoints can be taken into account, or a subset of them.
The whole capture can be used for synchronization, or you can choose a time range when the participant is roughly horizontally static but with a clear vertical motion (set approx_time_maxspeed
and time_range_around_maxspeed
accordingly).
N.B.: Works best when:
- the participant does not move towards or away from the cameras
- they perform a clear vertical movement
- the capture lasts at least 5 seconds long, so that there is enough data to synchronize on
- the capture lasts a few minutes maximum, so that cameras are less likely to drift with time
N.B.: Alternatively, synchronize cameras using a flashlight, a clap, or a clear event. GoPro cameras can also be synchronized with a timecode, by GPS (outdoors), or with their app (slightly less reliable).
Associate persons across cameras
If
multi_person
is set tofalse
, the algorithm chooses the person for whom the reprojection error is smallest.
Ifmulti_person
is set totrue
, it associates across views the people for whom the distances between epipolar lines are the smallest. People are then associated across frames according to their displacement speed.
N.B.: Skip this step if only one person is in the field of view.
Open an Anaconda prompt or a terminal in a Session
or Trial
folder.
Type ipython
.
from Pose2Sim import Pose2Sim
Pose2Sim.personAssociation()
Check printed output. If results are not satisfying, try and release the constraints in the Config.toml file.
Triangulating keypoints
Triangulate your 2D coordinates in a robust way.
The triangulation is weighted by the likelihood of each detected 2D keypoint, provided that they this likelihood is above a threshold.
If the reprojection error is above another threshold, right and left sides are swapped; if it is still above, cameras are removed until the threshold is met. If more cameras are removed than a predefined number, triangulation is skipped for this point and this frame. In the end, missing values are interpolated.
Open an Anaconda prompt or a terminal in a Session
or Trial
folder.
Type ipython
.
from Pose2Sim import Pose2Sim
Pose2Sim.triangulation()
Check printed output, and visualize your trc in OpenSim: File -> Preview experimental data
.
If your triangulation is not satisfying, try and release the constraints in the Config.toml
file.
Filtering 3D coordinates
Filter your 3D coordinates.
Butterworth, Kalman, Butterworth on speed, Gaussian, LOESS, Median filters are available and can be tuned accordingly.
Open an Anaconda prompt or a terminal in a Session
or Trial
folder.
Type ipython
.
from Pose2Sim import Pose2Sim
Pose2Sim.filtering()
Check your filtration with the displayed figures, and visualize your .trc file in OpenSim. If your filtering is not satisfying, try and change the parameters in the Config.toml
file.
Output:
Marker Augmentation
Use the Stanford LSTM model to estimate the position of 47 virtual markers.
Note that inverse kinematic results are not necessarily better after marker augmentation. Skip if results are not convincing.
N.B.: Marker augmentation tends to give a more stable, but less precise output. In practice, it is mostly beneficial when using less than 4 cameras.
Make sure that participant_height
is correct in your Config.toml
file. participant_mass
is mostly optional for IK.
Only works with models estimating at least the following keypoints (e.g., not COCO):
["Neck", "RShoulder", "LShoulder", "RHip", "LHip", "RKnee", "LKnee",
"RAnkle", "LAnkle", "RHeel", "LHeel", "RSmallToe", "LSmallToe",
"RBigToe", "LBigToe", "RElbow", "LElbow", "RWrist", "LWrist"]
Will not work properly if missing values are not interpolated (i.e., if there are Nan value in the .trc file).
Open an Anaconda prompt or a terminal in a Session
or Trial
folder.
Type ipython
.
from Pose2Sim import Pose2Sim
Pose2Sim.markerAugmentation()
OpenSim kinematics
Obtain a scaled model and 3D joint angles.
This can be either done fully automatically within Pose2Sim, or manually within OpenSim GUI.
Within Pose2Sim
Scaling and inverse kinematics are performed in a fully automatic way for each trc file.
No need for a static trial!
Note that automatic scaling is not recommended when the participant is mostly crouching or sitting. In this case, scale manually on a standing trial (see next section).
Model scaling is done according to the mean of the segment lengths, across a subset of frames. We remove the 10% fastest frames (potential outliers), the frames where the speed is 0 (person probably out of frame), and the 40% most extreme segment values (potential outliers).
In your Config.toml file, set use_augmentation = false
is you don't want to use the results with augmented marker (this is sometimes better).
Set right_left_symmetry = false
if you have good reasons to think the participant is not symmetrical (e.g. if they wear a prosthetic limb).
Open an Anaconda prompt or a terminal in a Session
or Trial
folder.
Type ipython
.
from Pose2Sim import Pose2Sim
Pose2Sim.kinematics()
Once you have the scaled model and the joint angles, you are free to go further! Inverse dynamics, muscle analysis, etc. (make sure previously add muscles from the Pose2Sim model with muscles).
Within OpenSim GUI
If you are not fully satisfied with the results or on sitting or crouching trials, you can perform scaling and inverse kinematics in a more traditional way, with (or without) a static trial.
Scaling
- Choose a time range where the 3D keypoints are particularly well reconstructed, or capture a static pose, typically an A-pose...
- Open OpenSim.
- File -> Open Model: Open the provided
Model_Pose2Sim_LSTM.osim
model fromPose2Sim/OpenSim_Setup
.
Note: Here and below, replace 'LSTM' by any other model if needed, e.g. HALPE_26 - Tools -> Scale model -> Load: Load the provided
Scaling_Setup_Pose2Sim_LSTM.xml
scaling file. - Replace the example .trc file with your own data.
- Run
- File > Save Model: Save the new scaled OpenSim model.
Inverse kinematics
- Tools -> Inverse kinematics -> Load: Load the provided
IK_Setup_Pose2Sim_LSTM.xml
scaling file fromPose2Sim/OpenSim_Setup
. - Replace the example .trc file with your own data, and specify the path to your angle kinematics output file.
- Run.
- Right click on the Model->Motions->Coordinates > Save As: Save angle results
Command line
Alternatively, you can use command-line tools:
import opensim
opensim.ScaleTool("<PATH TO YOUR SCALING OR IK SETUP FILE>.xml").run()
opensim.InverseKinematicsTool("<PATH TO YOUR SCALING OR IK SETUP FILE>.xml").run()
N.B.: You'll need to adjust the time_range
, output_motion_file
, and enter the absolute path (NOT the relative path) to the input and output .osim
, .trc
, and .mot
files in your setup file.
You can also run other API commands. See there for more instructions on how to use it.
Utilities
A list of standalone tools (see Utilities), which can be either run as scripts, or imported as functions. Check usage in the docstring of each Python file. The figure below shows how some of these tools can be used to further extend Pose2Sim usage.
Converting pose files (CLICK TO SHOW)
Blazepose_runsave.py Runs BlazePose on a video, and saves coordinates in OpenPose (json) or DeepLabCut (h5 or csv) format.
DLC_to_OpenPose.py Converts a DeepLabCut (h5) 2D pose estimation file into OpenPose (json) files.
AlphaPose_to_OpenPose.py Converts AlphaPose single json file to OpenPose frame-by-frame files.
Converting calibration files (CLICK TO SHOW)
calib_toml_to_easymocap.py Converts an OpenCV .toml calibration file to EasyMocap intrinsic and extrinsic .yml calibration files.
calib_easymocap_to_toml.py Converts EasyMocap intrinsic and extrinsic .yml calibration files to an OpenCV .toml calibration file.
calib_from_checkerboard.py Calibrates cameras with images or a video of a checkerboard, saves calibration in a Pose2Sim .toml calibration file. You should probably use Pose2Sim.calibration() instead, which is much easier and better.
calib_qca_to_toml.py Converts a Qualisys .qca.txt calibration file to the Pose2Sim .toml calibration file (similar to what is used in AniPose).
calib_toml_to_qca.py Converts a Pose2Sim .toml calibration file (e.g., from a checkerboard) to a Qualisys .qca.txt calibration file.
calib_toml_to_opencap.py Converts an OpenCV .toml calibration file to OpenCap .pickle calibration files.
calib_toml_to_opencap.py To convert OpenCap calibration tiles to a .toml file, please use Pose2Sim.calibration() and set convert_from = 'opencap' in Config.toml.
Plotting tools (CLICK TO SHOW)
json_display_with_img.py Overlays 2D detected json coordinates on original raw images. High confidence keypoints are green, low confidence ones are red.
json_display_without_img.py Plots an animation of 2D detected json coordinates.
trc_plot.py Displays X, Y, Z coordinates of each 3D keypoint of a TRC file in a different matplotlib tab.
Other trc tools (CLICK TO SHOW)
trc_from_easymocap.py Convert EasyMocap results keypoints3d .json files to .trc.
c3d_to_trc.py Converts 3D point data from a .c3d file to a .trc file compatible with OpenSim. No analog data (force plates, emg) nor computed data (angles, powers, etc.) are retrieved.
trc_to_c3d.py Converts 3D point data from a .trc file to a .c3d file compatible with Visual3D.
trc_desample.py Undersamples a trc file.
trc_Zup_to_Yup.py Changes Z-up system coordinates to Y-up system coordinates.
trc_filter.py Filters trc files. Available filters: Butterworth, Kalman, Butterworth on speed, Gaussian, LOESS, Median.
trc_gaitevents.py Detects gait events from point coordinates according to Zeni et al. (2008).
trc_combine.py Combine two trc files, for example a triangulated DeepLabCut trc file and a triangulated OpenPose trc file.
trc_from_mot_osim.py Build a trc file from a .mot motion file and a .osim model file.
bodykin_from_mot_osim.py Converts a mot file to a .csv file with rotation and orientation of all segments.
reproj_from_trc_calib.py Reprojects 3D coordinates of a trc file to the image planes defined by a calibration file. Output in OpenPose or DeepLabCut format.
How to cite and how to contribute
How to cite
If you use this code or data, please cite Pagnon et al., 2022b, Pagnon et al., 2022a, or Pagnon et al., 2021.
@Article{Pagnon_2022_JOSS,
AUTHOR = {Pagnon, David and Domalain, Mathieu and Reveret, Lionel},
TITLE = {Pose2Sim: An open-source Python package for multiview markerless kinematics},
JOURNAL = {Journal of Open Source Software},
YEAR = {2022},
DOI = {10.21105/joss.04362},
URL = {https://joss.theoj.org/papers/10.21105/joss.04362}
}
@Article{Pagnon_2022_Accuracy,
AUTHOR = {Pagnon, David and Domalain, Mathieu and Reveret, Lionel},
TITLE = {Pose2Sim: An End-to-End Workflow for 3D Markerless Sports Kinematics—Part 2: Accuracy},
JOURNAL = {Sensors},
YEAR = {2022},
DOI = {10.3390/s22072712},
URL = {https://www.mdpi.com/1424-8220/22/7/2712}
}
@Article{Pagnon_2021_Robustness,
AUTHOR = {Pagnon, David and Domalain, Mathieu and Reveret, Lionel},
TITLE = {Pose2Sim: An End-to-End Workflow for 3D Markerless Sports Kinematics—Part 1: Robustness},
JOURNAL = {Sensors},
YEAR = {2021},
DOI = {10.3390/s21196530},
URL = {https://www.mdpi.com/1424-8220/21/19/6530}
}
How to contribute and to-do list
I would happily welcome any proposal for new features, code improvement, and more!
If you want to contribute to Pose2Sim, please see this issue.
You will be proposed a to-do list, but please feel absolutely free to propose your own ideas and improvements.
Main to-do list
- Graphical User Interface
- Synchronization
- Self-calibration based on keypoint detection
Detailed GOT-DONE and TO-DO list (CLICK TO SHOW)
✔ Pose: Support OpenPose body_25b for more accuracy, body_135 for pronation/supination. ✔ Pose: Support BlazePose for faster inference (on mobile device). ✔ Pose: Support DeepLabCut for training on custom datasets. ✔ Pose: Support AlphaPose as an alternative to OpenPose. ✔ Pose: Define custom model in config.toml rather than in skeletons.py. ✔ Pose: Integrate pose estimation within Pose2Sim (via RTMlib). ▢ Pose: Support MMPose, SLEAP, etc. ▢ Pose: Implement RTMPoseW3D and monocular 3D kinematics ▢ Pose: Directly reading from DeepLabCut .csv or .h5 files instead of converting to .json (triangulation, person association, calibration, synchronization...) ▢ Pose: GUI help for DeepLabCut model creation.
✔ Calibration: Convert Qualisys .qca.txt calibration file. ✔ Calibration: Convert Optitrack extrinsic calibration file. ✔ Calibration: Convert Vicon .xcp calibration file. ✔ Calibration: Convert OpenCap .pickle calibration files. ✔ Calibration: Convert EasyMocap .yml calibration files. ✔ Calibration: Convert bioCV calibration files. ✔ Calibration: Easier and clearer calibration procedure: separate intrinsic and extrinsic parameter calculation, edit corner detection if some are wrongly detected (or not visible). ✔ Calibration: Possibility to evaluate extrinsic parameters from cues on scene. ▢ Calibration: Support vertical checkerboard. ▢ Calibration: Once object points have been detected or clicked once, track them for live calibration of moving cameras. Propose to click again when they are lost. ▢ Calibration: Calibrate cameras by pairs and compute average extrinsic calibration with aniposelib. ▢ Calibration: Fine-tune calibration with bundle adjustment. ▢ Calibration: Support ChArUco board detection (see there). ▢ Calibration: Calculate calibration with points rather than board. (1) SBA calibration with wand (cf Argus, see converter here). Set world reference frame in the end. ▢ Calibration: Alternatively, self-calibrate with OpenPose keypoints. Set world reference frame in the end. ▢ Calibration: Convert fSpy calibration based on vanishing point.
✔ Synchronization: Synchronize cameras on keypoint speeds. ▢ Synchronization: Synchronize in multi-person mode: click on the person to synchronize on.
✔ Person Association: Automatically choose the main person to triangulate. ✔ Person Association: Multiple persons association. 1. Triangulate all the persons whose reprojection error is below a certain threshold (instead of only the one with minimum error), and then track in time with speed cf Slembrouck 2020? or 2. Based on affinity matrices Dong 2021? or 3. Based on occupancy maps Yildiz 2012? or 4. With a neural network Huang 2023?
✔ Triangulation: Triangulation weighted with confidence. ✔ Triangulation: Set a likelihood threshold below which a camera should not be used, a reprojection error threshold, and a minimum number of remaining cameras below which triangulation is skipped for this frame. ✔ Triangulation: Interpolate missing frames (cubic, bezier, linear, slinear, quadratic) ✔ Triangulation: Show mean reprojection error in px and in mm for each keypoint. ✔ Triangulation: Show how many cameras on average had to be excluded for each keypoint. ✔ Triangulation: Evaluate which cameras were the least reliable. ✔ Triangulation: Show which frames had to be interpolated for each keypoint. ✔ Triangulation: Solve limb swapping (although not really an issue with Body_25b). Try triangulating with opposite side if reprojection error too large. Alternatively, ignore right and left sides, use RANSAC or SDS triangulation, and then choose right or left by majority voting. More confidence can be given to cameras whose plane is the most coplanar to the right/left line. ✔ Triangulation: Undistort 2D points before triangulating (and distort them before computing reprojection error). ✔ Triangulation: Offer the possibility to augment the triangulated data with the OpenCap LSTM. Create "BODY_25_AUGMENTED" model, Scaling_setup, IK_Setup. ✔ Triangulation: Multiple person kinematics (output multiple .trc coordinates files). Triangulate all persons with reprojection error above threshold, and identify them by minimizing their displacement across frames. ▢ Triangulation: Pre-compile weighted_triangulation and reprojection with @jit(nopython=True, parallel=True) for faster execution. ▢ Triangulation: Offer the possibility of triangulating with Sparse Bundle Adjustment (SBA), Extended Kalman Filter (EKF), Full Trajectory Estimation (FTE) (see AcinoSet). ▢ Triangulation: Implement normalized DLT and RANSAC triangulation, Outlier rejection (sliding z-score?), as well as a triangulation refinement step. ▢ Triangulation: Track hands and face, and add articulated OpenSim hand.
✔ Filtering: Available filtering methods: Butterworth, Butterworth on speed, Gaussian, Median, LOESS (polynomial smoothing). ✔ Filtering: Implement Kalman filter and Kalman smoother. ▢ Filtering: Implement smoothNet
✔ OpenSim: Integrate better spine from lifting fullbody model to the gait full-body model, more accurate for the knee. ✔ OpenSim: Optimize model marker positions as compared to ground-truth marker-based positions. ✔ OpenSim: Add scaling and inverse kinematics setup files. ✔ OpenSim: Add full model with contact spheres (SmoothSphereHalfSpaceForce) and full-body muscles (DeGrooteFregly2016Muscle), for Moco for example. ✔ OpenSim: Add model with ISB shoulder. ✔ OpenSim: Integrate OpenSim in Pose2Sim. ✔ OpenSim: Do not require a separate scaling trial: scale on the 10% slowest frames of the moving trial instead, or take median scaling value. ▢ OpenSim: Implement optimal fixed-interval Kalman smoothing for inverse kinematics (this OpenSim fork), or Biorbd)
✔ GUI: Blender add-on (cf MPP2SOS), Maya-Mocap and BlendOsim. ▢ GUI: App or webapp (e.g., with gradio, Streamlit, or Napari ). Also see tkinter interfaces (or Kivy if we want something nice and portable, or Google Colab). Maybe have a look at the Deeplabcut GUI for inspiration. ▢ GUI: 3D plot of cameras and of triangulated keypoints. ▢ GUI: Demo on Google Colab (see Sports2D for OpenPose and Python package installation on Google Drive).
✔ Demo: Provide Demo data for users to test the code. ✔ Demo: Add videos for users to experiment with other pose detection 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, MkDocs, or github.io (maybe better) for clearer documentation.
✔ Pip package ✔ Batch processing (also enable non-batch processing) ✔ Catch errors ▢ Conda package ▢ Docker image ▢ Integrate Sports2D for OpenSim analysis from a single camera ▢ Real-time: Run Pose estimation, Person association, Triangulation, Kalman filter, IK frame by frame (instead of running each step for all frames) ▢ Config parameter for non batch peocessing
▢ 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: Create trc_to_c3d.py script
✔ Bug: calibration.py. FFMPEG error message when calibration files are images. See there. ✔ Bug: common.py, class plotWindow(). Python crashes after a few runs of
Pose2Sim.filtering()
whendisplay_figures=true
. See there.
Acknowledgements:
- Supervised my PhD: @lreveret (INRIA, Université Grenoble Alpes), and @mdomalai (Université de Poitiers).
- Provided the Demo data: @aaiaueil from Université Gustave Eiffel.
- Tested the code and provided feedback: @simonozan, @daeyongyang, @ANaaim, @rlagnsals
- Submitted various accepted pull requests: @ANaaim, @rlagnsals, @peterlololsss
- Provided a code snippet for Optitrack calibration: @claraaudap (Université Bretagne Sud).
- Issued MPP2SOS, a (non-free) Blender extension based on Pose2Sim: @carlosedubarreto
BSD 3-Clause License
Copyright (c) 2022, perfanalytics All rights reserved.
Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met:
-
Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer.
-
Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution.
-
Neither the name of the copyright holder nor the names of its contributors may be used to endorse or promote products derived from this software without specific prior written permission.
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
File details
Details for the file pose2sim-0.10.4.tar.gz
.
File metadata
- Download URL: pose2sim-0.10.4.tar.gz
- Upload date:
- Size: 36.2 MB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.9.20
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 24cb23563350555386b4c248198b7057d76b18ac88838d0ab88a3419efa7ffb6 |
|
MD5 | 212e2c9828b6bb8aaaca5b1f3bacb65a |
|
BLAKE2b-256 | 154c3af0868d139791708c764d896d64aaebadf7cddc631b95381ee840be7294 |
File details
Details for the file pose2sim-0.10.4-py3-none-any.whl
.
File metadata
- Download URL: pose2sim-0.10.4-py3-none-any.whl
- Upload date:
- Size: 36.3 MB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.9.20
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 4c055b58347f557a00fbd2dad483470d89bec50df569270d6469c737974c8da5 |
|
MD5 | 08fad5145dd0038e24dce74267cfb74c |
|
BLAKE2b-256 | ad05c2d4adf2b5be3e46339b0655003bc86940277ed2050154d1b9e419ddbb5e |