support CMU Mocap

2021-08-11 12:20:38 +08:00
parent 5562416dc6
commit 2963a13933
1 changed files with 4 additions and 1 deletions
--- a/fit/tools/label.py
+++ b/fit/tools/label.py
@ -1166,4 +1166,7 @@ def get_label(file_name, dataset_name):
        return UTD_MHAD[key]
    elif dataset_name == 'CMU_Mocap':
        key = file_name.split('.')[0]
-        return CMU_Mocap[key]
+        if key in CMU_Mocap.keys():
+            return CMU_Mocap[key]
+        else:
+            return ""