|
{ |
|
"_name_or_path": "nvidia/mit-b0", |
|
"architectures": [ |
|
"SegformerForSemanticSegmentation" |
|
], |
|
"attention_probs_dropout_prob": 0.0, |
|
"classifier_dropout_prob": 0.1, |
|
"decoder_hidden_size": 256, |
|
"depths": [ |
|
2, |
|
2, |
|
2, |
|
2 |
|
], |
|
"downsampling_rates": [ |
|
1, |
|
4, |
|
8, |
|
16 |
|
], |
|
"drop_path_rate": 0.1, |
|
"hidden_act": "gelu", |
|
"hidden_dropout_prob": 0.0, |
|
"hidden_sizes": [ |
|
32, |
|
64, |
|
160, |
|
256 |
|
], |
|
"id2label": { |
|
"0": "background", |
|
"1": "candy", |
|
"2": "egg tart", |
|
"3": "french fries", |
|
"4": "chocolate", |
|
"5": "biscuit", |
|
"6": "popcorn", |
|
"7": "pudding", |
|
"8": "ice cream", |
|
"9": "cheese butter", |
|
"10": "cake", |
|
"11": "wine", |
|
"12": "milkshake", |
|
"13": "coffee", |
|
"14": "juice", |
|
"15": "milk", |
|
"16": "tea", |
|
"17": "almond", |
|
"18": "red beans", |
|
"19": "cashew", |
|
"20": "dried cranberries", |
|
"21": "soy", |
|
"22": "walnut", |
|
"23": "peanut", |
|
"24": "egg", |
|
"25": "apple", |
|
"26": "date", |
|
"27": "apricot", |
|
"28": "avocado", |
|
"29": "banana", |
|
"30": "strawberry", |
|
"31": "cherry", |
|
"32": "blueberry", |
|
"33": "raspberry", |
|
"34": "mango", |
|
"35": "olives", |
|
"36": "peach", |
|
"37": "lemon", |
|
"38": "pear", |
|
"39": "fig", |
|
"40": "pineapple", |
|
"41": "grape", |
|
"42": "kiwi", |
|
"43": "melon", |
|
"44": "orange", |
|
"45": "watermelon", |
|
"46": "steak", |
|
"47": "pork", |
|
"48": "chicken duck", |
|
"49": "sausage", |
|
"50": "fried meat", |
|
"51": "lamb", |
|
"52": "sauce", |
|
"53": "crab", |
|
"54": "fish", |
|
"55": "shellfish", |
|
"56": "shrimp", |
|
"57": "soup", |
|
"58": "bread", |
|
"59": "corn", |
|
"60": "hamburg", |
|
"61": "pizza", |
|
"62": " hanamaki baozi", |
|
"63": "wonton dumplings", |
|
"64": "pasta", |
|
"65": "noodles", |
|
"66": "rice", |
|
"67": "pie", |
|
"68": "tofu", |
|
"69": "eggplant", |
|
"70": "potato", |
|
"71": "garlic", |
|
"72": "cauliflower", |
|
"73": "tomato", |
|
"74": "kelp", |
|
"75": "seaweed", |
|
"76": "spring onion", |
|
"77": "rape", |
|
"78": "ginger", |
|
"79": "okra", |
|
"80": "lettuce", |
|
"81": "pumpkin", |
|
"82": "cucumber", |
|
"83": "white radish", |
|
"84": "carrot", |
|
"85": "asparagus", |
|
"86": "bamboo shoots", |
|
"87": "broccoli", |
|
"88": "celery stick", |
|
"89": "cilantro mint", |
|
"90": "snow peas", |
|
"91": " cabbage", |
|
"92": "bean sprouts", |
|
"93": "onion", |
|
"94": "pepper", |
|
"95": "green beans", |
|
"96": "French beans", |
|
"97": "king oyster mushroom", |
|
"98": "shiitake", |
|
"99": "enoki mushroom", |
|
"100": "oyster mushroom", |
|
"101": "white button mushroom", |
|
"102": "salad", |
|
"103": "other ingredients" |
|
}, |
|
"image_size": 224, |
|
"initializer_range": 0.02, |
|
"label2id": { |
|
" cabbage": 91, |
|
" hanamaki baozi": 62, |
|
"French beans": 96, |
|
"almond": 17, |
|
"apple": 25, |
|
"apricot": 27, |
|
"asparagus": 85, |
|
"avocado": 28, |
|
"background": 0, |
|
"bamboo shoots": 86, |
|
"banana": 29, |
|
"bean sprouts": 92, |
|
"biscuit": 5, |
|
"blueberry": 32, |
|
"bread": 58, |
|
"broccoli": 87, |
|
"cake": 10, |
|
"candy": 1, |
|
"carrot": 84, |
|
"cashew": 19, |
|
"cauliflower": 72, |
|
"celery stick": 88, |
|
"cheese butter": 9, |
|
"cherry": 31, |
|
"chicken duck": 48, |
|
"chocolate": 4, |
|
"cilantro mint": 89, |
|
"coffee": 13, |
|
"corn": 59, |
|
"crab": 53, |
|
"cucumber": 82, |
|
"date": 26, |
|
"dried cranberries": 20, |
|
"egg": 24, |
|
"egg tart": 2, |
|
"eggplant": 69, |
|
"enoki mushroom": 99, |
|
"fig": 39, |
|
"fish": 54, |
|
"french fries": 3, |
|
"fried meat": 50, |
|
"garlic": 71, |
|
"ginger": 78, |
|
"grape": 41, |
|
"green beans": 95, |
|
"hamburg": 60, |
|
"ice cream": 8, |
|
"juice": 14, |
|
"kelp": 74, |
|
"king oyster mushroom": 97, |
|
"kiwi": 42, |
|
"lamb": 51, |
|
"lemon": 37, |
|
"lettuce": 80, |
|
"mango": 34, |
|
"melon": 43, |
|
"milk": 15, |
|
"milkshake": 12, |
|
"noodles": 65, |
|
"okra": 79, |
|
"olives": 35, |
|
"onion": 93, |
|
"orange": 44, |
|
"other ingredients": 103, |
|
"oyster mushroom": 100, |
|
"pasta": 64, |
|
"peach": 36, |
|
"peanut": 23, |
|
"pear": 38, |
|
"pepper": 94, |
|
"pie": 67, |
|
"pineapple": 40, |
|
"pizza": 61, |
|
"popcorn": 6, |
|
"pork": 47, |
|
"potato": 70, |
|
"pudding": 7, |
|
"pumpkin": 81, |
|
"rape": 77, |
|
"raspberry": 33, |
|
"red beans": 18, |
|
"rice": 66, |
|
"salad": 102, |
|
"sauce": 52, |
|
"sausage": 49, |
|
"seaweed": 75, |
|
"shellfish": 55, |
|
"shiitake": 98, |
|
"shrimp": 56, |
|
"snow peas": 90, |
|
"soup": 57, |
|
"soy": 21, |
|
"spring onion": 76, |
|
"steak": 46, |
|
"strawberry": 30, |
|
"tea": 16, |
|
"tofu": 68, |
|
"tomato": 73, |
|
"walnut": 22, |
|
"watermelon": 45, |
|
"white button mushroom": 101, |
|
"white radish": 83, |
|
"wine": 11, |
|
"wonton dumplings": 63 |
|
}, |
|
"layer_norm_eps": 1e-06, |
|
"mlp_ratios": [ |
|
4, |
|
4, |
|
4, |
|
4 |
|
], |
|
"model_type": "segformer", |
|
"num_attention_heads": [ |
|
1, |
|
2, |
|
5, |
|
8 |
|
], |
|
"num_channels": 3, |
|
"num_encoder_blocks": 4, |
|
"patch_sizes": [ |
|
7, |
|
3, |
|
3, |
|
3 |
|
], |
|
"reshape_last_stage": true, |
|
"semantic_loss_ignore_index": 255, |
|
"sr_ratios": [ |
|
8, |
|
4, |
|
2, |
|
1 |
|
], |
|
"strides": [ |
|
4, |
|
2, |
|
2, |
|
2 |
|
], |
|
"torch_dtype": "float32", |
|
"transformers_version": "4.46.3" |
|
} |
|
|