From cd66285690b827ad62769b49013a8a8c9e6010ff Mon Sep 17 00:00:00 2001 From: davidpagnon Date: Sun, 9 Jun 2024 12:05:42 +0200 Subject: [PATCH] does not try to augment more trc than len(participant_mass) --- Pose2Sim/markerAugmentation.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/Pose2Sim/markerAugmentation.py b/Pose2Sim/markerAugmentation.py index 9b9fe79..d125486 100644 --- 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.