This commit is contained in:
lmd
2025-07-25 15:05:31 +08:00
parent 24e1e31234
commit 65769e5eb6
18 changed files with 3642 additions and 67 deletions

View File

@ -1284,8 +1284,9 @@ def get_label(file_name, dataset_name):
key = file_name[-5:]
return HumanAct12[key]
elif dataset_name == 'UTD_MHAD':
key = file_name.split('_')[0][1:]
return UTD_MHAD[key]
###key = file_name.split('_')[0][1:]
###return UTD_MHAD[key]
return "single_person"
elif dataset_name == 'CMU_Mocap':
key = file_name.split(':')[0]
return CMU_Mocap[key] if key in CMU_Mocap.keys() else ""