From 2963a13933c2bd6cbd18286e7601f0d39f7df9b9 Mon Sep 17 00:00:00 2001
From: Iridoudou <2534936416@qq.com>
Date: Wed, 11 Aug 2021 12:20:38 +0800
Subject: [PATCH] support CMU Mocap

---
 fit/tools/label.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/fit/tools/label.py b/fit/tools/label.py
index 0c40021..de6eea7 100644
--- 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 ""