id2class.py

# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
# SPDX-License-Identifier: Apache-2.0

id2class_map = {
    "candle": {"0": "BACKGROUND", "1": "Chunk Of Wax Missing", "2": "Damaged Corner of Packaging",
               "3": "weird candle wick", "4": "Different Colour Spot", "5": "Extra Wax in Candle",
               "6": "Foreign Particals on Candle", "7": "Wax Melded out of the candle", "8": "Other"},
    "capsules": {"0": "BACKGROUND", "1": "Bubble", "2": "Discolor", "3": "Scratch", "4": "Leak", "5": "Misshape"},
    "fryum": {"0": "BACKGROUND", "1": "Burnt", "2": "Corner or Edge Breakage", "3": "Different Colour Spot",
              "4": "Fryum Stuck Together", "5": "Middle Breakage", "6": "Similar Colour Spot", "7": "Small Scratches",
              "8": "Other"},
    "cashew": {"0": "BACKGROUND", "1": "Burnt", "2": "Corner or Edge Breakage", "3": "Different Colour Spot",
               "4": "Middle Breakage", "5": "Small Holes", "6": "Small Scratches", "7": "Stuck Together",
               "8": "Same Colour Spot", "9": "Other"},
    "chewinggum": {"0": "BACKGROUND", "1": "Chunk of gum missing", "2": "Corner Missing", "3": "Scratches",
                   "4": "Similar Colour Spot", "5": "Small Cracks", "6": "Other"},
    "macaroni1": {"0": "BACKGROUND", "1": "Chip Around Edge And Corner", "2": "Different Colour Spot",
                  "3": "Middle Breakage", "4": "Similar Colour Spot", "5": "Small Cracks", "6": "Small Scratches",
                  "7": "Other"},
    "macaroni2": {"0": "BACKGROUND", "1": "Breakage down the middle", "2": "Color spot similar to the Object",
                  "3": "Different Color spot", "4": "Small chip around edge", "5": "Small Cracks",
                  "6": "Small Scratches", "7": "Other"},
    "pcb1": {"0": "BACKGROUND", "1": "Bent", "2": "Melt", "3": "Scratch", "4": "Missing"},
    "pcb2": {"0": "BACKGROUND", "1": "Bent", "2": "Melt", "3": "Scratch", "4": "Missing"},
    "pcb3": {"0": "BACKGROUND", "1": "Bent", "2": "Melt", "3": "Scratch", "4": "Missing"},
    "pcb4": {"0": "BACKGROUND", "1": "Burnt", "2": "Scratch", "3": "Missing", "4": "Damage", "5": "Extra",
             "6": "Wrong Place", "7": "Dirt"},
    "pipe_fryum": {"0": "BACKGROUND", "1": "Burnt", "2": "Corner And Edge Breakage", "3": "Different Colour Spot",
                   "4": "Middle Breakage", "5": "Similar Colour Spot", "6": "Small Scratches", "7": "Stuck Together",
                   "8": "Small Cracks", "9": "Other"}
}