fulyvv/pose2sim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Process non filtered data if there is no data filtered in folder (#86)

Browse Source
This commit is contained in:
Alexandre Naaim 2024-04-02 11:15:48 +02:00 committed by GitHub
parent b21846a834
commit 71e05dd48d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
Pose2Sim/markerAugmentation.py
View File

@ -79,9 +79,18 @@ def augmentTRC(config_dict):
offset = True
# Apply all trc files
trc_files = [f for f in glob.glob(os.path.join(pathInputTRCFile, '*.trc')) if 'filt' in f and '_LSTM' not in f]
if len(trc_files) == 0:
raise ValueError('No filtered trc files found.')
all_trc_files = [f for f in glob.glob(os.path.join(pathInputTRCFile, '*.trc')) if '_LSTM' not in f]
trc_no_filtering = [f for f in glob.glob(os.path.join(pathInputTRCFile, '*.trc')) if
'_LSTM' not in f and 'filt' not in f]
trc_filtering = [f for f in glob.glob(os.path.join(pathInputTRCFile, '*.trc')) if '_LSTM' not in f and 'filt' in f]
if len(all_trc_files) == 0:
raise ValueError('No trc files found.')
if len(trc_filtering) > 0:
trc_files = trc_filtering
else:
trc_files = trc_no_filtering
for p, pathInputTRCFile in enumerate(trc_files):
pathOutputTRCFile = os.path.splitext(pathInputTRCFile)[0] + '_LSTM.trc'