pose2sim/Pose2Sim/Demo/S00_Demo_Session/test.py

39 lines
1.1 KiB
Python

import os
import toml
def test_workflow():
from Pose2Sim import Pose2Sim
# Calibration
config_dict = toml.load('Config.toml')
config_dict.get("project").update({"project_dir":"."})
Pose2Sim.calibration(config_dict)
# Static trial
project_dir = os.path.join("S00_P00_Participant","S00_P00_T00_StaticTrial")
config_dict.get("project").update({"project_dir":project_dir})
config_dict['filtering']['display_figures'] = False
# Pose2Sim.poseEstimation(config_dict)
# Pose2Sim.synchronization(config_dict)
Pose2Sim.personAssociation(config_dict)
Pose2Sim.triangulation(config_dict)
Pose2Sim.filtering(config_dict)
# Pose2Sim.kinematics(config_dict)
# Balancing trial
project_dir = os.path.join("S00_P00_Participant","S00_P00_T01_BalancingTrial")
config_dict.get("project").update({"project_dir":project_dir})
config_dict['filtering']['display_figures'] = False
# Pose2Sim.poseEstimation(config_dict)
# Pose2Sim.synchronization(config_dict)
Pose2Sim.personAssociation(config_dict)
Pose2Sim.triangulation(config_dict)
Pose2Sim.filtering(config_dict)
# Pose2Sim.kinematics(config_dict)