name
This commit is contained in:
@ -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 ""
|
||||
|
||||
Reference in New Issue
Block a user