saving format changed to pkl

2021-08-09 11:00:12 +08:00
parent 1a5646d843
commit c3dc13d2c8
3 changed files with 88 additions and 8 deletions
--- a/fit/tools/label.py
+++ b/fit/tools/label.py
@ -0,0 +1,75 @@
+HumanAct12 = {
+    "A0101": "warm_up_wristankle",
+    "A0102": "warm_up_pectoral",
+    "A0103": "warm_up_eblowback",
+    "A0104": "warm_up_bodylean_right_arm",
+    "A0105": "warm_up_bodylean_left_arm",
+    "A0106": "warm_up_bow_right",
+    "A0107": "warm_up_bow_left",
+    "A0201": "walk",
+    "A0301": "run",
+    "A0401": "jump_handsup",
+    "A0402": "jump_vertical",
+    "A0501": "drink_bottle_righthand",
+    "A0502": "drink_bottle_lefthand",
+    "A0503": "drink_cup_righthand",
+    "A0504": "drink_cup_lefthand",
+    "A0505": "drink_both_hands",
+    "A0601": "lift_dumbbell with _right hand",
+    "A0602": "lift_dumbbell with _left hand",
+    "A0603": "Lift dumbells with both hands",
+    "A0604": "lift_dumbbell over head",
+    "A0605": "lift_dumbells with both hands and bend legs",
+    "A0701": "sit",
+    "A0801": "eat_finger_right",
+    "A0802": "eat_pie/hamburger",
+    "A0803": "Eat with left hand",
+    "A0901": "Turn steering wheel",
+    "A1001": "Take out phone, call and put phone back",
+    "A1002": "Call with left hand",
+    "A1101": "boxing_left_right",
+    "A1102": "boxing_left_upwards",
+    "A1103": "boxing_right_upwards",
+    "A1104": "boxing_right_left",
+    "A1201": "throw_right_hand",
+    "A1202": "throw_both_hands"
+}
+
+UTD_MHAD = {
+    "1": "  right arm swipe to the left(swipt_left)",
+    "2": "  right arm swipe to the right(swipt_right)",
+    "3": "  right hand wave(wave)",
+    "4": "  two hand front clap(clap)",
+    "5": "  right arm throw(throw)",
+    "6": "  cross arms in the chest(arm_cross)",
+    "7": "  basketball shooting(basketball_shoot)",
+    "8": "  draw x(draw_x)",
+    "9": "  draw circle(clockwise)(draw_circle_CW)",
+    "10": " draw circle(counter clockwise)(draw_circle_CCW)",
+    "11": " draw triangle(draw_triangle)",
+    "12": " bowling(right hand)(bowling)",
+    "13": " front boxing(boxing)",
+    "14": " baseball swing from right(baseball_swing)",
+    "15": " tennis forehand swing(tennis_swing)",
+    "16": " arm curl(two arms)(arm_curl)",
+    "17": " tennis serve(tennis_serve)",
+    "18": " two hand push(push)",
+    "19": " knock on door(knock)",
+    "20": " hand catch(catch)",
+    "21": " pick up and throw(pickup_throw)",
+    "22": " jogging(jog)",
+    "23": " walking(walk)",
+    "24": " sit to stand(sit2stand)",
+    "25": " stand to sit(stand2sit)",
+    "26": " forward lunge(left foot forward)(lunge)",
+    "27": " squat(squat)"
+}
+
+
+def get_label(file_name, dataset_name):
+    if dataset_name == 'HumanAct12':
+        key = file_name[-5:]
+        return HumanAct12[key]
+    elif dataset_name == 'UTD_MHAD':
+        key = file_name.split('_')[0][1:]
+        return UTD_MHAD[key]