epic-detr-latest / config.json
hussamturjman's picture
Upload DetrForObjectDetection
8f0a393
{
"activation_dropout": 0.0,
"activation_function": "relu",
"architectures": [
"DetrForObjectDetection"
],
"attention_dropout": 0.0,
"auxiliary_loss": false,
"backbone": "resnet50",
"bbox_cost": 5,
"bbox_loss_coefficient": 5,
"class_cost": 1,
"d_model": 256,
"decoder_attention_heads": 8,
"decoder_ffn_dim": 2048,
"decoder_layerdrop": 0.0,
"decoder_layers": 6,
"dice_loss_coefficient": 1,
"dilation": false,
"dropout": 0.1,
"encoder_attention_heads": 8,
"encoder_ffn_dim": 2048,
"encoder_layerdrop": 0.0,
"encoder_layers": 6,
"eos_coefficient": 0.1,
"giou_cost": 2,
"giou_loss_coefficient": 2,
"id2label": {
"0": "bag",
"1": "bin",
"2": "bottle",
"3": "carrot",
"4": "carrots",
"5": "container",
"6": "courgette",
"7": "cup",
"8": "cupboard",
"9": "cutlery",
"10": "cutting board",
"11": "door",
"12": "drawer",
"13": "fork",
"14": "fridge",
"15": "glass",
"16": "hob",
"17": "kettle",
"18": "knife",
"19": "lid",
"20": "oil",
"21": "onion",
"22": "pan",
"23": "plate",
"24": "rice",
"25": "salt",
"26": "sauce pan",
"27": "saucepan",
"28": "spatula",
"29": "spoon",
"30": "table",
"31": "tap",
"32": "vegetables",
"33": "cloth",
"34": "colander",
"35": "cooker",
"36": "counter",
"37": "liquid",
"38": "rag",
"39": "sink",
"40": "sponge",
"41": "washing",
"42": "box",
"43": "bowl",
"44": "bread",
"45": "food",
"46": "frying pan",
"47": "ladle",
"48": "mushroom",
"49": "plastic",
"50": "potato",
"51": "potatoes",
"52": "sauce",
"53": "sausage",
"54": "skin",
"55": "aubergine",
"56": "dough",
"57": "flour",
"58": "leaves",
"59": "olives",
"60": "onions",
"61": "oven",
"62": "pizza",
"63": "plates",
"64": "scissors",
"65": "tray",
"66": "lettuce",
"67": "salad",
"68": "tomatoes",
"69": "pepper",
"70": "squash",
"71": "coffee",
"72": "coffee pot",
"73": "filter",
"74": "milk",
"75": "mug",
"76": "pot",
"77": "cheese",
"78": "chicken",
"79": "chopping board",
"80": "microwave",
"81": "pasta",
"82": "pesto",
"83": "stove",
"84": "tupperware",
"85": "jar",
"86": "cucumber",
"87": "packaging",
"88": "tomato",
"89": "garlic",
"90": "meat",
"91": "mixture",
"92": "peel",
"93": "spice",
"94": "drying rack",
"95": "dishwasher",
"96": "egg",
"97": "eggs",
"98": "patty",
"99": "paneer",
"100": "ginger",
"101": "spices",
"102": "curry",
"103": "pack",
"104": "package",
"105": "teaspoon",
"106": "ingredients",
"107": "bacon",
"108": "chilli",
"109": "dish",
"110": "tortilla",
"111": "kitchen",
"112": "salad bowl",
"113": "surface",
"114": "cabinet",
"115": "fish",
"116": "tortillas",
"117": "peach"
},
"init_std": 0.02,
"init_xavier_std": 1.0,
"is_encoder_decoder": true,
"label2id": {
"aubergine": 55,
"bacon": 107,
"bag": 0,
"bin": 1,
"bottle": 2,
"bowl": 43,
"box": 42,
"bread": 44,
"cabinet": 114,
"carrot": 3,
"carrots": 4,
"cheese": 77,
"chicken": 78,
"chilli": 108,
"chopping board": 79,
"cloth": 33,
"coffee": 71,
"coffee pot": 72,
"colander": 34,
"container": 5,
"cooker": 35,
"counter": 36,
"courgette": 6,
"cucumber": 86,
"cup": 7,
"cupboard": 8,
"curry": 102,
"cutlery": 9,
"cutting board": 10,
"dish": 109,
"dishwasher": 95,
"door": 11,
"dough": 56,
"drawer": 12,
"drying rack": 94,
"egg": 96,
"eggs": 97,
"filter": 73,
"fish": 115,
"flour": 57,
"food": 45,
"fork": 13,
"fridge": 14,
"frying pan": 46,
"garlic": 89,
"ginger": 100,
"glass": 15,
"hob": 16,
"ingredients": 106,
"jar": 85,
"kettle": 17,
"kitchen": 111,
"knife": 18,
"ladle": 47,
"leaves": 58,
"lettuce": 66,
"lid": 19,
"liquid": 37,
"meat": 90,
"microwave": 80,
"milk": 74,
"mixture": 91,
"mug": 75,
"mushroom": 48,
"oil": 20,
"olives": 59,
"onion": 21,
"onions": 60,
"oven": 61,
"pack": 103,
"package": 104,
"packaging": 87,
"pan": 22,
"paneer": 99,
"pasta": 81,
"patty": 98,
"peach": 117,
"peel": 92,
"pepper": 69,
"pesto": 82,
"pizza": 62,
"plastic": 49,
"plate": 23,
"plates": 63,
"pot": 76,
"potato": 50,
"potatoes": 51,
"rag": 38,
"rice": 24,
"salad": 67,
"salad bowl": 112,
"salt": 25,
"sauce": 52,
"sauce pan": 26,
"saucepan": 27,
"sausage": 53,
"scissors": 64,
"sink": 39,
"skin": 54,
"spatula": 28,
"spice": 93,
"spices": 101,
"sponge": 40,
"spoon": 29,
"squash": 70,
"stove": 83,
"surface": 113,
"table": 30,
"tap": 31,
"teaspoon": 105,
"tomato": 88,
"tomatoes": 68,
"tortilla": 110,
"tortillas": 116,
"tray": 65,
"tupperware": 84,
"vegetables": 32,
"washing": 41
},
"mask_loss_coefficient": 1,
"max_position_embeddings": 1024,
"model_type": "detr",
"num_channels": 3,
"num_hidden_layers": 6,
"num_queries": 100,
"position_embedding_type": "sine",
"scale_embedding": false,
"torch_dtype": "float32",
"transformers_version": "4.23.1",
"use_pretrained_backbone": true
}