157 lines
4.6 KiB
Python
157 lines
4.6 KiB
Python
#! /usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
|
|
'''
|
|
########################################
|
|
## Pose2Sim tests ##
|
|
########################################
|
|
|
|
- BATCH SESSION:
|
|
- Calibration conversion from .qca.txt
|
|
- Single person:
|
|
- synchronization
|
|
- person association
|
|
- triangulation
|
|
- filtering
|
|
- marker augmentation
|
|
- Multi-person:
|
|
- synchronization
|
|
- person association
|
|
- triangulation
|
|
- filtering
|
|
- marker augmentation
|
|
|
|
- SINGLE TRIAL:
|
|
- calibration
|
|
- synchronization
|
|
- person association
|
|
- triangulation
|
|
- filtering
|
|
- marker augmentation
|
|
|
|
N.B.:
|
|
1. Calibration from scene dimensions is not tested, as it requires the
|
|
user to click points on the image.
|
|
2. OpenSim scaling and IK are not tested yet
|
|
3. Not all possible configuration parameters are extensively tested.
|
|
|
|
Usage:
|
|
cd Pose2Sim/Utilities
|
|
python tests.py
|
|
'''
|
|
|
|
## INIT
|
|
import os
|
|
import toml
|
|
from unittest.mock import patch
|
|
import unittest
|
|
|
|
from Pose2Sim import Pose2Sim
|
|
|
|
|
|
## AUTHORSHIP INFORMATION
|
|
__author__ = "David Pagnon"
|
|
__copyright__ = "Copyright 2021, Pose2Sim"
|
|
__credits__ = ["David Pagnon"]
|
|
__license__ = "BSD 3-Clause License"
|
|
__version__ = "0.9.4"
|
|
__maintainer__ = "David Pagnon"
|
|
__email__ = "contact@david-pagnon.com"
|
|
__status__ = "Development"
|
|
|
|
|
|
## FUNCTIONS
|
|
class TestWorkflow(unittest.TestCase):
|
|
@patch('builtins.input', return_value='no') # Mock input() to return 'no'
|
|
def test_workflow(self, mock_input):
|
|
'''
|
|
SINGLE-PERSON, MULTI-PERSON, BATCH PROCESSING:
|
|
- calibration
|
|
- pose estimation
|
|
- synchronization
|
|
- person association
|
|
- triangulation
|
|
- filtering
|
|
- marker augmentation
|
|
- run all
|
|
|
|
N.B.: Calibration from scene dimensions is not tested, as it requires the
|
|
user to click points on the image.
|
|
Not all possible configuration parameters are extensively tested.
|
|
Scaling and inverse kinematics are not tested yet.
|
|
|
|
Usage:
|
|
from Pose2Sim import tests; tests.test_workflow()
|
|
python tests.py
|
|
'''
|
|
|
|
|
|
###################
|
|
# SINGLE-PERSON #
|
|
###################
|
|
|
|
project_dir = '../Demo_SinglePerson'
|
|
config_dict = toml.load(os.path.join(project_dir, 'Config.toml'))
|
|
|
|
os.chdir(project_dir)
|
|
config_dict.get("project").update({"project_dir":project_dir})
|
|
config_dict.get("pose").update({"mode":'lightweight'})
|
|
config_dict.get("pose").update({"display_detection":False})
|
|
config_dict.get("synchronization").update({"display_sync_plots":False})
|
|
# config_dict['triangulation']['reorder_trc'] = False
|
|
config_dict['filtering']['display_figures'] = False
|
|
|
|
Pose2Sim.calibration(config_dict)
|
|
Pose2Sim.poseEstimation(config_dict)
|
|
Pose2Sim.synchronization(config_dict)
|
|
Pose2Sim.personAssociation(config_dict)
|
|
Pose2Sim.triangulation(config_dict)
|
|
Pose2Sim.filtering(config_dict)
|
|
Pose2Sim.markerAugmentation(config_dict)
|
|
# Pose2Sim.kinematics(config_dict)
|
|
|
|
config_dict.get("pose").update({"overwrite_pose":False})
|
|
Pose2Sim.runAll(config_dict)
|
|
|
|
|
|
####################
|
|
# MULTI-PERSON #
|
|
####################
|
|
|
|
project_dir = '../Demo_MultiPerson'
|
|
config_dict = toml.load(os.path.join(project_dir, 'Config.toml'))
|
|
|
|
os.chdir(project_dir)
|
|
config_dict.get("project").update({"project_dir":project_dir})
|
|
config_dict.get("pose").update({"mode":'lightweight'})
|
|
config_dict.get("pose").update({"display_detection":False})
|
|
config_dict.get("synchronization").update({"display_sync_plots":False})
|
|
config_dict['filtering']['display_figures'] = False
|
|
|
|
Pose2Sim.calibration(config_dict)
|
|
Pose2Sim.poseEstimation(config_dict)
|
|
Pose2Sim.synchronization(config_dict)
|
|
Pose2Sim.personAssociation(config_dict)
|
|
Pose2Sim.triangulation(config_dict)
|
|
Pose2Sim.filtering(config_dict)
|
|
Pose2Sim.markerAugmentation(config_dict)
|
|
# Pose2Sim.kinematics(config_dict)
|
|
|
|
config_dict.get("pose").update({"overwrite_pose":False})
|
|
Pose2Sim.runAll(config_dict)
|
|
|
|
|
|
####################
|
|
# BATCH PROCESSING #
|
|
####################
|
|
|
|
project_dir = '../Demo_Batch'
|
|
os.chdir(project_dir)
|
|
|
|
Pose2Sim.runAll()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|