does not try to augment more trc than len(participant_mass)

2024-06-09 12:05:42 +02:00 · 2024-06-09 12:05:42 +02:00 · cd66285690
commit cd66285690
parent 34781a855a
1 changed files with 2 additions and 1 deletions
--- a/Pose2Sim/markerAugmentation.py
+++ b/Pose2Sim/markerAugmentation.py
@ -109,7 +109,8 @@ def augmentTRC(config_dict):
    else:
        trc_files = trc_no_filtering

-    for p, pathInputTRCFile in enumerate(trc_files):
+    for p in range(len(subject_mass)):
+        pathInputTRCFile = trc_files[p]
        pathOutputTRCFile = os.path.splitext(pathInputTRCFile)[0] + '_LSTM.trc'
    
        # This is by default - might need to be adjusted in the future.