get augmenterDir from site-packages

2024-03-01 09:59:55 +01:00 · 2024-03-01 09:59:55 +01:00 · 757673d01c
commit 757673d01c
parent fcae1ca02e
1 changed files with 1 additions and 1 deletions
--- a/Pose2Sim/markerAugmentation.py
+++ b/Pose2Sim/markerAugmentation.py
@ -76,7 +76,7 @@ def augmentTRC(config_dict):
    if not type(subject_height) == list:
        subject_height = [subject_height]
        subject_mass = [subject_mass]
-    augmenterDir = os.path.join(session_dir, '..', 'MarkerAugmenter')
+    augmenterDir = os.path.dirname(utilsDataman.__file__)
    augmenterModelName = 'LSTM'
    augmenter_model = 'v0.3'
    offset = True