Update config.json

config.json

@@ -1,167 +1,167 @@
-{
-  "id2label": {
-    "0": "person",
-    "1": "bicycle",
-    "2": "car",
-    "3": "motorcycle",
-    "4": "airplane",
-    "5": "bus",
-    "6": "train",
-    "7": "truck",
-    "8": "boat",
-    "9": "traffic light",
-    "10": "fire hydrant",
-    "11": "stop sign",
-    "12": "parking meter",
-    "13": "bench",
-    "14": "bird",
-    "15": "cat",
-    "16": "dog",
-    "17": "horse",
-    "18": "sheep",
-    "19": "cow",
-    "20": "elephant",
-    "21": "bear",
-    "22": "zebra",
-    "23": "giraffe",
-    "24": "backpack",
-    "25": "umbrella",
-    "26": "handbag",
-    "27": "tie",
-    "28": "suitcase",
-    "29": "frisbee",
-    "30": "skis",
-    "31": "snowboard",
-    "32": "sports ball",
-    "33": "kite",
-    "34": "baseball bat",
-    "35": "baseball glove",
-    "36": "skateboard",
-    "37": "surfboard",
-    "38": "tennis racket",
-    "39": "bottle",
-    "40": "wine glass",
-    "41": "cup",
-    "42": "fork",
-    "43": "knife",
-    "44": "spoon",
-    "45": "bowl",
-    "46": "banana",
-    "47": "apple",
-    "48": "sandwich",
-    "49": "orange",
-    "50": "broccoli",
-    "51": "carrot",
-    "52": "hot dog",
-    "53": "pizza",
-    "54": "donut",
-    "55": "cake",
-    "56": "chair",
-    "57": "couch",
-    "58": "potted plant",
-    "59": "bed",
-    "60": "dining table",
-    "61": "toilet",
-    "62": "tv",
-    "63": "laptop",
-    "64": "mouse",
-    "65": "remote",
-    "66": "keyboard",
-    "67": "cell phone",
-    "68": "microwave",
-    "69": "oven",
-    "70": "toaster",
-    "71": "sink",
-    "72": "refrigerator",
-    "73": "book",
-    "74": "clock",
-    "75": "vase",
-    "76": "scissors",
-    "77": "teddy bear",
-    "78": "hair drier",
-    "79": "toothbrush"
-  },
-  "label2id": {
-    "person": 0,
-    "bicycle": 1,
-    "car": 2,
-    "motorcycle": 3,
-    "airplane": 4,
-    "bus": 5,
-    "train": 6,
-    "truck": 7,
-    "boat": 8,
-    "traffic light": 9,
-    "fire hydrant": 10,
-    "stop sign": 11,
-    "parking meter": 12,
-    "bench": 13,
-    "bird": 14,
-    "cat": 15,
-    "dog": 16,
-    "horse": 17,
-    "sheep": 18,
-    "cow": 19,
-    "elephant": 20,
-    "bear": 21,
-    "zebra": 22,
-    "giraffe": 23,
-    "backpack": 24,
-    "umbrella": 25,
-    "handbag": 26,
-    "tie": 27,
-    "suitcase": 28,
-    "frisbee": 29,
-    "skis": 30,
-    "snowboard": 31,
-    "sports ball": 32,
-    "kite": 33,
-    "baseball bat": 34,
-    "baseball glove": 35,
-    "skateboard": 36,
-    "surfboard": 37,
-    "tennis racket": 38,
-    "bottle": 39,
-    "wine glass": 40,
-    "cup": 41,
-    "fork": 42,
-    "knife": 43,
-    "spoon": 44,
-    "bowl": 45,
-    "banana": 46,
-    "apple": 47,
-    "sandwich": 48,
-    "orange": 49,
-    "broccoli": 50,
-    "carrot": 51,
-    "hot dog": 52,
-    "pizza": 53,
-    "donut": 54,
-    "cake": 55,
-    "chair": 56,
-    "couch": 57,
-    "potted plant": 58,
-    "bed": 59,
-    "dining table": 60,
-    "toilet": 61,
-    "tv": 62,
-    "laptop": 63,
-    "mouse": 64,
-    "remote": 65,
-    "keyboard": 66,
-    "cell phone": 67,
-    "microwave": 68,
-    "oven": 69,
-    "toaster": 70,
-    "sink": 71,
-    "refrigerator": 72,
-    "book": 73,
-    "clock": 74,
-    "vase": 75,
-    "scissors": 76,
-    "teddy bear": 77,
-    "hair drier": 78,
-    "toothbrush": 79
-  },
-  "model_type": "yolos"
+# New id2label with English + Chinese translations
+id2label_translated = {
+    "0": "person 人",
+    "1": "bicycle 自行车",
+    "2": "car 汽车",
+    "3": "motorcycle 摩托车",
+    "4": "airplane 飞机",
+    "5": "bus 公共汽车",
+    "6": "train 火车",
+    "7": "truck 卡车",
+    "8": "boat 船",
+    "9": "traffic light 交通灯",
+    "10": "fire hydrant 消防栓",
+    "11": "stop sign 停止标志",
+    "12": "parking meter 停车计时器",
+    "13": "bench 长凳",
+    "14": "bird 鸟",
+    "15": "cat 猫",
+    "16": "dog 狗",
+    "17": "horse 马",
+    "18": "sheep 羊",
+    "19": "cow 牛",
+    "20": "elephant 大象",
+    "21": "bear 熊",
+    "22": "zebra 斑马",
+    "23": "giraffe 长颈鹿",
+    "24": "backpack 背包",
+    "25": "umbrella 雨伞",
+    "26": "handbag 手提包",
+    "27": "tie 领带",
+    "28": "suitcase 手提箱",
+    "29": "frisbee 飞盘",
+    "30": "skis 滑雪板",
+    "31": "snowboard 单板滑雪",
+    "32": "sports ball 运动球",
+    "33": "kite 风筝",
+    "34": "baseball bat 棒球棒",
+    "35": "baseball glove 棒球手套",
+    "36": "skateboard 滑板",
+    "37": "surfboard 冲浪板",
+    "38": "tennis racket 网球拍",
+    "39": "bottle 瓶子",
+    "40": "wine glass 酒杯",
+    "41": "cup 杯子",
+    "42": "fork 叉子",
+    "43": "knife 刀",
+    "44": "spoon 勺子",
+    "45": "bowl 碗",
+    "46": "banana 香蕉",
+    "47": "apple 苹果",
+    "48": "sandwich 三明治",
+    "49": "orange 橙子",
+    "50": "broccoli 西兰花",
+    "51": "carrot 胡萝卜",
+    "52": "hot dog 热狗",
+    "53": "pizza 披萨",
+    "54": "donut 甜甜圈",
+    "55": "cake 蛋糕",
+    "56": "chair 椅子",
+    "57": "couch 沙发",
+    "58": "potted plant 盆栽",
+    "59": "bed 床",
+    "60": "dining table 餐桌",
+    "61": "toilet 马桶",
+    "62": "tv 电视",
+    "63": "laptop 笔记本电脑",
+    "64": "mouse 鼠标",
+    "65": "remote 遥控器",
+    "66": "keyboard 键盘",
+    "67": "cell phone 手机",
+    "68": "microwave 微波炉",
+    "69": "oven 烤箱",
+    "70": "toaster 烤面包机",
+    "71": "sink 水槽",
+    "72": "refrigerator 冰箱",
+    "73": "book 书",
+    "74": "clock 时钟",
+    "75": "vase 花瓶",
+    "76": "scissors 剪刀",
+    "77": "teddy bear 泰迪熊",
+    "78": "hair drier 吹风机",
+    "79": "toothbrush 牙刷"
+}
+
+# New label2id with English + Chinese translations
+label2id_translated = {
+    "person 人": 0,
+    "bicycle 自行车": 1,
+    "car 汽车": 2,
+    "motorcycle 摩托车": 3,
+    "airplane 飞机": 4,
+    "bus 公共汽车": 5,
+    "train 火车": 6,
+    "truck 卡车": 7,
+    "boat 船": 8,
+    "traffic light 交通灯": 9,
+    "fire hydrant 消防栓": 10,
+    "stop sign 停止标志": 11,
+    "parking meter 停车计时器": 12,
+    "bench 长凳": 13,
+    "bird 鸟": 14,
+    "cat 猫": 15,
+    "dog 狗": 16,
+    "horse 马": 17,
+    "sheep 羊": 18,
+    "cow 牛": 19,
+    "elephant 大象": 20,
+    "bear 熊": 21,
+    "zebra 斑马": 22,
+    "giraffe 长颈鹿": 23,
+    "backpack 背包": 24,
+    "umbrella 雨伞": 25,
+    "handbag 手提包": 26,
+    "tie 领带": 27,
+    "suitcase 手提箱": 28,
+    "frisbee 飞盘": 29,
+    "skis 滑雪板": 30,
+    "snowboard 单板滑雪": 31,
+    "sports ball 运动球": 32,
+    "kite 风筝": 33,
+    "baseball bat 棒球棒": 34,
+    "baseball glove 棒球手套": 35,
+    "skateboard 滑板": 36,
+    "surfboard 冲浪板": 37,
+    "tennis racket 网球拍": 38,
+    "bottle 瓶子": 39,
+    "wine glass 酒杯": 40,
+    "cup 杯子": 41,
+    "fork 叉子": 42,
+    "knife 刀": 43,
+    "spoon 勺子": 44,
+    "bowl 碗": 45,
+    "banana 香蕉": 46,
+    "apple 苹果": 47,
+    "sandwich 三明治": 48,
+    "orange 橙子": 49,
+    "broccoli 西兰花": 50,
+    "carrot 胡萝卜": 51,
+    "hot dog 热狗": 52,
+    "pizza 披萨": 53,
+    "donut 甜甜圈": 54,
+    "cake 蛋糕": 55,
+    "chair 椅子": 56,
+    "couch 沙发": 57,
+    "potted plant 盆栽": 58,
+    "bed 床": 59,
+    "dining table 餐桌": 60,
+    "toilet 马桶": 61,
+    "tv 电视": 62,
+    "laptop 笔记本电脑": 63,
+    "mouse 鼠标": 64,
+    "remote 遥控器": 65,
+    "keyboard 键盘": 66,
+    "cell phone 手机": 67,
+    "microwave 微波炉": 68,
+    "oven 烤箱": 69,
+    "toaster 烤面包机": 70,
+    "sink 水槽": 71,
+    "refrigerator 冰箱": 72,
+    "book 书": 73,
+    "clock 时钟": 74,
+    "vase 花瓶": 75,
+    "scissors 剪刀": 76,
+    "teddy bear 泰迪熊": 77,
+    "hair drier 吹风机": 78,
+    "toothbrush 牙刷": 79
 }