Update

backup/kinetics_400.json

@@ -0,0 +1,402 @@
+{
+    "0": "đi xuống",
+    "1": "tiếng trống không khí",
+    "2": "trả lời câu hỏi",
+    "3": "vỗ tay",
+    "4": "bôi kem",
+    "5": "bắn cung",
+    "6": "vật tay",
+    "7": "cắm hoa",
+    "8": "máy tính lắp ráp",
+    "9": "đấu giá",
+    "10": "em bé thức dậy",
+    "11": "nướng bánh quy",
+    "12": "thổi bóng bay",
+    "13": "băng bó",
+    "14": "đồ nướng",
+    "15": "pha chế",
+    "16": "beatbox",
+    "17": "nuôi ong",
+    "18": "múa bụng",
+    "19": "bấm băng ghế dự bị",
+    "20": "cong lưng",
+    "21": "kim loại uốn",
+    "22": "đạp xe qua tuyết",
+    "23": "phun cát",
+    "24": "thổi thủy tinh",
+    "25": "thổi lá",
+    "26": "xì mũi",
+    "27": "thổi nến",
+    "28": "trượt ván",
+    "29": "đóng sách",
+    "30": "nảy trên tấm bạt lò xo",
+    "31": "chơi bowling",
+    "32": "tóc tết",
+    "33": "bánh mì hoặc vụn bánh mì",
+    "34": "nhảy đột phá",
+    "35": "tranh cọ",
+    "36": "chải tóc",
+    "37": "đánh răng",
+    "38": "tủ xây dựng",
+    "39": "nhà kho",
+    "40": "nhảy bungee",
+    "41": "đi hát rong",
+    "42": "chèo thuyền hoặc chèo thuyền kayak",
+    "43": "capoeira",
+    "44": "bế con",
+    "45": "đu quay",
+    "46": "khắc bí ngô",
+    "47": "bắt cá",
+    "48": "bắt hoặc ném bóng chày",
+    "49": "bắt hoặc ném dĩa nhựa",
+    "50": "bắt hoặc ném bóng mềm",
+    "51": "ăn mừng",
+    "52": "thay dầu",
+    "53": "bánh xe thay đổi",
+    "54": "kiểm tra lốp xe",
+    "55": "cổ vũ",
+    "56": "chặt củi",
+    "57": "vỗ tay",
+    "58": "làm đồ gốm bằng đất sét",
+    "59": "sạch và giật",
+    "60": "lau sàn",
+    "61": "làm sạch máng xối",
+    "62": "vệ sinh hồ bơi",
+    "63": "làm sạch giày",
+    "64": "dọn nhà vệ sinh",
+    "65": "làm sạch cửa sổ",
+    "66": "leo dây",
+    "67": "leo thang",
+    "68": "leo cây",
+    "69": "tung hứng tiếp xúc",
+    "70": "nấu gà",
+    "71": "nấu trứng",
+    "72": "nấu ăn trên lửa trại",
+    "73": "nấu xúc xích",
+    "74": "đếm tiền",
+    "75": "nhảy dòng đồng quê",
+    "76": "bẻ cổ",
+    "77": "em bé đang bò",
+    "78": "vượt sông",
+    "79": "khóc",
+    "80": "tóc uốn",
+    "81": "cắt móng tay",
+    "82": "cắt dứa",
+    "83": "cắt dưa hấu",
+    "84": "múa ba lê",
+    "85": "khiêu vũ charleston",
+    "86": "nhảy gangnam style",
+    "87": "macarena nhảy múa",
+    "88": "cử tạ",
+    "89": "trang trí cây thông Noel",
+    "90": "đào",
+    "91": "ăn uống",
+    "92": "chơi gôn đĩa",
+    "93": "vách đá lặn",
+    "94": "bóng né",
+    "95": "tập thể dục nhịp điệu",
+    "96": "giặt đồ",
+    "97": "làm nail",
+    "98": "vẽ",
+    "99": "rê bóng rổ",
+    "100": "uống",
+    "101": "uống bia",
+    "102": "uống rượu",
+    "103": "lái xe",
+    "104": "lái máy kéo",
+    "105": "thả đá",
+    "106": "ngón tay đánh trống",
+    "107": "nhúng bóng rổ",
+    "108": "tóc sắp chết",
+    "109": "ăn burger",
+    "110": "ăn bánh",
+    "111": "ăn cà rốt",
+    "112": "ăn khoai tây chiên",
+    "113": "ăn bánh rán",
+    "114": "ăn xúc xích",
+    "115": "ăn kem",
+    "116": "ăn spaghetti",
+    "117": "ăn dưa hấu",
+    "118": "săn trứng",
+    "119": "tập thể dục cánh tay",
+    "120": "tập thể dục với bóng tập thể dục",
+    "121": "dập lửa",
+    "122": "cấy mặt",
+    "123": "cho chim ăn",
+    "124": "cho cá ăn",
+    "125": "cho dê ăn",
+    "126": "làm đầy lông mày",
+    "127": "búng ngón tay",
+    "128": "sửa tóc",
+    "129": "bánh kếp lật",
+    "130": "thả diều",
+    "131": "gấp quần áo",
+    "132": "gấp khăn ăn",
+    "133": "gấp giấy",
+    "134": "tăng trước",
+    "135": "chiên rau củ",
+    "136": "thu gom rác",
+    "137": "súc miệng",
+    "138": "cắt tóc",
+    "139": "xăm mình",
+    "140": "trao hoặc nhận giải thưởng",
+    "141": "đánh gôn",
+    "142": "lái xe gôn",
+    "143": "đánh gôn",
+    "144": "xay thịt",
+    "145": "chó chải chuốt",
+    "146": "ngựa chải chuốt",
+    "147": "thể dục nhào lộn",
+    "148": "ném búa",
+    "149": "đập đầu",
+    "150": "cú húc đầu",
+    "151": "nhảy cao",
+    "152": "cú đá cao",
+    "153": "đánh bóng chày",
+    "154": "điểm dừng khúc côn cầu",
+    "155": "ôm rắn",
+    "156": "nhảy lò cò",
+    "157": "lướt ván",
+    "158": "ôm",
+    "159": "hula hooping",
+    "160": "vượt rào",
+    "161": "vượt rào (thể thao)",
+    "162": "leo băng",
+    "163": "câu cá trên băng",
+    "164": "trượt băng",
+    "165": "ủi",
+    "166": "ném lao",
+    "167": "trượt ván",
+    "168": "chạy bộ",
+    "169": "tung hứng bóng",
+    "170": "tung hứng lửa",
+    "171": "tung hứng bóng đá",
+    "172": "nhảy vào hồ bơi",
+    "173": "nhảy kiểu nhảy",
+    "174": "khung thành trên sân",
+    "175": "đá bóng",
+    "176": "hôn nhau",
+    "177": "lướt sóng",
+    "178": "đan",
+    "179": "tiếng kêu",
+    "180": "cười",
+    "181": "đặt gạch",
+    "182": "nhảy xa",
+    "183": "lung tung",
+    "184": "làm bánh",
+    "185": "làm bánh sandwich",
+    "186": "dọn giường",
+    "187": "làm trang sức",
+    "188": "làm bánh pizza",
+    "189": "làm người tuyết",
+    "190": "làm sushi",
+    "191": "pha trà",
+    "192": "hành quân",
+    "193": "mát xa lưng",
+    "194": "mát xa chân",
+    "195": "mát xa chân",
+    "196": "người xoa bóp đầu",
+    "197": "vắt sữa bò",
+    "198": "lau sàn",
+    "199": "mô tô",
+    "200": "di chuyển đồ đạc",
+    "201": "cắt cỏ",
+    "202": "tin tức mới nhất",
+    "203": "mở chai",
+    "204": "quà khai trương",
+    "205": "dù lượn",
+    "206": "dù lượn",
+    "207": "parkour",
+    "208": "vượt qua bóng bầu dục Mỹ (trong trò chơi)",
+    "209": "vượt qua bóng bầu dục Mỹ (không trong trò chơi)",
+    "210": "gọt vỏ táo",
+    "211": "gọt vỏ khoai tây",
+    "212": "nuôi thú cưng (không phải mèo)",
+    "213": "mèo cưng",
+    "214": "hái quả",
+    "215": "trồng cây",
+    "216": "trát",
+    "217": "chơi đàn accordion",
+    "218": "chơi cầu lông",
+    "219": "chơi kèn túi",
+    "220": "chơi bóng rổ",
+    "221": "chơi guitar bass",
+    "222": "chơi bài",
+    "223": "chơi đàn Cello",
+    "224": "chơi cờ",
+    "225": "chơi kèn clarinet",
+    "226": "chơi điều khiển",
+    "227": "chơi cricket",
+    "228": "chơi chũm chọe",
+    "229": "chơi didgeridoo",
+    "230": "chơi trống",
+    "231": "thổi sáo",
+    "232": "chơi ghi-ta",
+    "233": "chơi kèn hòa tấu",
+    "234": "chơi đàn hạc",
+    "235": "chơi khúc côn cầu trên băng",
+    "236": "chơi bàn phím",
+    "237": "chơi đá bóng",
+    "238": "chơi độc quyền",
+    "239": "chơi đàn organ",
+    "240": "chơi súng sơn",
+    "241": "chơi piano",
+    "242": "chơi bài xì phé",
+    "243": "chơi ghi âm",
+    "244": "chơi saxophone",
+    "245": "chơi bóng quần hoặc bóng vợt",
+    "246": "chơi quần vợt",
+    "247": "chơi trombone",
+    "248": "chơi kèn",
+    "249": "chơi đàn ukulele",
+    "250": "chơi vĩ cầm",
+    "251": "chơi bóng chuyền",
+    "252": "chơi đàn xylophone",
+    "253": "hầm sào",
+    "254": "trình bày dự báo thời tiết",
+    "255": "kéo lên",
+    "256": "nắm tay bơm",
+    "257": "bơm xăng",
+    "258": "bao đấm",
+    "259": "người đấm bốc (đấm bốc)",
+    "260": "đẩy lên",
+    "261": "đẩy xe",
+    "262": "đẩy xe",
+    "263": "đẩy xe lăn",
+    "264": "đọc sách",
+    "265": "đọc báo",
+    "266": "ghi âm nhạc",
+    "267": "đi xe đạp",
+    "268": "cưỡi lạc đà",
+    "269": "cưỡi voi",
+    "270": "cưỡi bò cơ khí",
+    "271": "đạp xe leo núi",
+    "272": "cưỡi la",
+    "273": "cưỡi hoặc đi bộ với ngựa",
+    "274": "đi xe tay ga",
+    "275": "đi xe đạp một bánh",
+    "276": "xé giấy",
+    "277": "rô-bốt nhảy",
+    "278": "leo núi",
+    "279": "oẳn tù tì",
+    "280": "trượt patin",
+    "281": "chạy trên máy chạy bộ",
+    "282": "chèo thuyền",
+    "283": "nhảy salsa",
+    "284": "sàn chà nhám",
+    "285": "trứng bác",
+    "286": "lặn biển",
+    "287": "đặt bàn",
+    "288": "bắt tay",
+    "289": "lắc đầu",
+    "290": "mài dao",
+    "291": "gọt bút chì",
+    "292": "cạo đầu",
+    "293": "cạo lông chân",
+    "294": "xén lông cừu",
+    "295": "đánh giày",
+    "296": "bắn bóng rổ",
+    "297": "bóng đá (bóng đá)",
+    "298": "bắn đặt",
+    "299": "xẻng tuyết",
+    "300": "xé giấy",
+    "301": "xáo bài",
+    "302": "đá bên",
+    "303": "thông dịch ngôn ngữ ký hiệu",
+    "304": "hát",
+    "305": "ngồi dậy",
+    "306": "trượt ván",
+    "307": "nhảy trượt tuyết",
+    "308": "trượt tuyết (không phải slalom hoặc việt dã)",
+    "309": "trượt tuyết xuyên quốc gia",
+    "310": "trượt tuyết slalom",
+    "311": "nhảy dây",
+    "312": "nhảy dù",
+    "313": "chậm chạp",
+    "314": "tát",
+    "315": "đua chó kéo xe",
+    "316": "hút thuốc",
+    "317": "hút hookah",
+    "318": "cử tạ",
+    "319": "hắt hơi",
+    "320": "đánh hơi",
+    "321": "lặn biển",
+    "322": "trượt tuyết",
+    "323": "trượt tuyết",
+    "324": "di chuyển trên tuyết",
+    "325": "lộn nhào",
+    "326": "quay poi",
+    "327": "phun sơn",
+    "328": "phun",
+    "329": "lặn bàn đạp",
+    "330": "ngồi xổm",
+    "331": "thè lưỡi",
+    "332": "dậm nho",
+    "333": "duỗi tay",
+    "334": "duỗi chân",
+    "335": "gảy đàn guitar",
+    "336": "đám đông lướt sóng",
+    "337": "lướt nước",
+    "338": "quét sàn",
+    "339": "bơi ngửa",
+    "340": "bơi sải ngực",
+    "341": "bơi bướm",
+    "342": "nhảy đu",
+    "343": "đu đưa chân",
+    "344": "đu trên cái gì đó",
+    "345": "đấu kiếm",
+    "346": "thái cực quyền",
+    "347": "vua tắm",
+    "348": "nhảy tango",
+    "349": "nhảy tap",
+    "350": "gõ ghita",
+    "351": "bút gõ",
+    "352": "nếm bia",
+    "353": "nếm thức ăn",
+    "354": "làm chứng",
+    "355": "nhắn tin",
+    "356": "ném rìu",
+    "357": "ném bóng",
+    "358": "ném đĩa",
+    "359": "nhột",
+    "360": "trượt băng",
+    "361": "tung đồng xu",
+    "362": "ném salad",
+    "363": "chó huấn luyện",
+    "364": "đu dây",
+    "365": "cạo râu",
+    "366": "cắt tỉa cây",
+    "367": "nhảy ba lần",
+    "368": "thắt nơ",
+    "369": "thắt nút (không buộc)",
+    "370": "thắt cà vạt",
+    "371": "mở hộp",
+    "372": "xe tải dỡ hàng",
+    "373": "sử dụng máy tính",
+    "374": "sử dụng điều khiển từ xa (không chơi game)",
+    "375": "sử dụng segway",
+    "376": "kho tiền",
+    "377": "xếp hàng chờ",
+    "378": "dắt chó đi dạo",
+    "379": "rửa bát đĩa",
+    "380": "rửa chân",
+    "381": "gội đầu",
+    "382": "rửa tay",
+    "383": "trượt nước",
+    "384": "trượt nước",
+    "385": "tưới cây",
+    "386": "sáp trở lại",
+    "387": "tẩy lông ngực",
+    "388": "tẩy lông mày",
+    "389": "tẩy lông chân",
+    "390": "dệt giỏ",
+    "391": "hàn",
+    "392": "huýt sáo",
+    "393": "lướt ván",
+    "394": "gói quà",
+    "395": "đấu vật",
+    "396": "viết",
+    "397": "ngáp",
+    "398": "yoga",
+    "399": "zumba"
+}

run_opencv.py

@@ -35,7 +35,7 @@ class ArgParser(Tap):
 
     id2label: Optional[str] = "labels/kinetics_400.json"
 
-    threshold: Optional[float] = None # 10.0
+    threshold: Optional[float] = None  # 10.0
 
     max_confidence: Optional[float] = 20.0  # Set None if not scale
 
@@ -83,6 +83,8 @@ class ActivityModel:
 
         # model predicts one of the 400 Kinetics-400 classes
         max_index = logits.argmax(-1).item()
+        if max_index not in self.model.config.id2label:
+            return
         predicted_label = self.model.config.id2label[max_index]
 
         confidence = logits[0][max_index]
@@ -93,17 +95,17 @@ class ActivityModel:
             img_container.frame_rate.label = f"{predicted_label}_{confidence:.2f}%"
 
         # logits = np.squeeze(logits)
-        logits = logits.squeeze().numpy()
-        indices = np.argsort(logits)[::-1][: self.args.top_k]
-        values = logits[indices]
+        # logits = logits.squeeze().numpy()
+        # indices = np.argsort(logits)[::-1][: self.args.top_k]
+        # values = logits[indices]
 
-        results: List[Tuple[str, float]] = []
-        for index, value in zip(indices, values):
-            predicted_label = self.model.config.id2label[index]
-            # print(f"Label: {predicted_label} - {value:.2f}%")
-            results.append((predicted_label, value))
+        # results: List[Tuple[str, float]] = []
+        # for index, value in zip(indices, values):
+        #     predicted_label = self.model.config.id2label[index]
+        #     # print(f"Label: {predicted_label} - {value:.2f}%")
+        #     results.append((predicted_label, value))
 
-        img_container.rs = pd.DataFrame(results, columns=("Label", "Confidence"))
+        # img_container.rs = pd.DataFrame(results, columns=("Label", "Confidence"))
 
     def get_frames_per_video(self, model_name: str) -> int:
         if "base-finetuned" in model_name: