constants.py

label_map = {
    "n02106662": "dog",
    "n02124075": "cat",
    "n02281787": "butterfly",
    "n02389026": "horse",
    "n02492035": "monkey",
    "n02504458": "elephant",
    "n02510455": "panda or pandas",
    "n02607072": "clownfish or fish",
    "n02690373": "airplane",
    "n02906734": "brush",
    "n02951358": "canoe or boat",
    "n02992529": "cell phone",
    "n03063599": "coffee mug",
    "n03100240": "car",
    "n03180011": "computer",
    "n03197337": "watch or watches",
    "n03272010": "guitar",
    "n03272562": "train",
    "n03297495": "coffee maker",
    "n03376595": "chair",
    "n03445777": "ball",
    "n03452741": "piano",
    "n03584829": "ironing machine",
    "n03590841": "pumpkin",
    "n03709823": "handbag",
    "n03773504": "rocket",
    "n03775071": "gloves",
    "n03792782": "bicycle",
    "n03792972": "camp or camping",
    "n03877472": "girl",
    "n03888257": "parachutes",
    "n03982430": "pool table",
    "n04044716": "tower",
    "n04069434": "camera",
    "n04086273": "gun",
    "n04120489": "shoes",
    "n07753592": "fruits and vegetables",
    "n07873807": "pizza",
    "n11939491": "flower",
    "n13054560": "mushroom",
}

id2label = {"11": "butterfly", "17": "computer", "26": "girl", "19": "pizza", "21": "monkey", "7": "gun", "16": "ball", "36": "airplane", "1": "parachutes", "15": "cell phone", "31": "camp or camping", "29": "camera", "39": "cat", "3": "clownfish or fish", "37": "train", "38": "tower", "23": "handbag", "14": "pumpkin", "13": "shoes", "32": "fruits and vegetables", "12": "dog", "24": "car", "30": "piano", "8": "panda or pandas", "5": "coffee mug", "20": "rocket", "9": "flower", "2": "ironing machine", "4": "coffee maker", "28": "guitar", "35": "elephant", "10": "canoe or boat", "6": "bicycle", "33": "mushroom", "22": "pool table", "34": "watch or watches", "0": "horse", "27": "gloves", "25": "chair", "18": "brush"}