pose2sim/Pose2Sim/Demo/S00_Demo_Session/test.py

41 lines
1.2 KiB
Python
Raw Normal View History

2023-12-18 04:03:01 +08:00
import os
2023-12-06 16:48:11 +08:00
import toml
def test_workflow():
2023-12-17 21:29:22 +08:00
2023-12-06 16:48:11 +08:00
from Pose2Sim import Pose2Sim
2023-12-17 21:29:22 +08:00
2023-12-18 04:03:01 +08:00
# Calibration
2023-12-17 21:29:22 +08:00
config_dict = toml.load('Config.toml')
config_dict.get("project").update({"project_dir":"."})
2023-12-06 16:48:11 +08:00
Pose2Sim.calibration(config_dict)
2023-12-17 21:29:22 +08:00
# Static trial
2023-12-18 04:03:01 +08:00
project_dir = os.path.join("S00_P00_Participant","S00_P00_T00_StaticTrial")
config_dict.get("project").update({"project_dir":project_dir})
2023-12-17 21:29:22 +08:00
config_dict['filtering']['display_figures'] = False
2023-12-18 04:03:01 +08:00
2023-12-17 21:29:22 +08:00
# Pose2Sim.poseEstimation(config_dict)
# Pose2Sim.synchronization(config_dict)
2023-12-06 16:48:11 +08:00
Pose2Sim.personAssociation(config_dict)
Pose2Sim.triangulation(config_dict)
Pose2Sim.filtering(config_dict)
2024-01-20 03:03:35 +08:00
Pose2Sim.markerAugmentation(config_dict)
2023-12-17 21:29:22 +08:00
# Pose2Sim.kinematics(config_dict)
2023-12-18 04:03:01 +08:00
# 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
2023-12-18 04:03:01 +08:00
# Pose2Sim.poseEstimation(config_dict)
# Pose2Sim.synchronization(config_dict)
Pose2Sim.personAssociation(config_dict)
Pose2Sim.triangulation(config_dict)
Pose2Sim.filtering(config_dict)
2024-01-20 03:03:35 +08:00
Pose2Sim.markerAugmentation(config_dict)
# Pose2Sim.kinematics(config_dict)