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
|
|
|
|
2024-01-03 03:15:43 +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
|
|
|
|
2024-01-03 03:15:43 +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)
|
2023-12-17 21:29:22 +08:00
|
|
|
# Pose2Sim.kinematics(config_dict)
|
|
|
|
|
|
2023-12-18 04:03:01 +08:00
|
|
|
|
2024-01-05 11:20:12 +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
|
|
|
|
2024-01-05 11:20:12 +08:00
|
|
|
# Pose2Sim.poseEstimation(config_dict)
|
|
|
|
|
# Pose2Sim.synchronization(config_dict)
|
|
|
|
|
Pose2Sim.personAssociation(config_dict)
|
|
|
|
|
Pose2Sim.triangulation(config_dict)
|
|
|
|
|
Pose2Sim.filtering(config_dict)
|
|
|
|
|
# Pose2Sim.kinematics(config_dict)
|
