Image Segmentation
Transformers
PyTorch
upernet
Inference Endpoints
test2 / config.json
mccaly's picture
Upload 3 files
e0bbe93
{
"_commit_hash": null,
"architectures": [
"UperNetForSemanticSegmentation"
],
"auxiliary_channels": 256,
"auxiliary_concat_input": false,
"auxiliary_in_channels": 384,
"auxiliary_loss_weight": 0.4,
"auxiliary_num_convs": 1,
"backbone_config": {
"_name_or_path": "",
"add_cross_attention": false,
"architectures": null,
"attention_probs_dropout_prob": 0.0,
"bad_words_ids": null,
"begin_suppress_tokens": null,
"bos_token_id": null,
"chunk_size_feed_forward": 0,
"cross_attention_hidden_size": null,
"decoder_start_token_id": null,
"depths": [
2,
2,
18,
2
],
"diversity_penalty": 0.0,
"do_sample": false,
"drop_path_rate": 0.1,
"early_stopping": false,
"embed_dim": 96,
"encoder_no_repeat_ngram_size": 0,
"encoder_stride": 32,
"eos_token_id": null,
"exponential_decay_length_penalty": null,
"finetuning_task": null,
"forced_bos_token_id": null,
"forced_eos_token_id": null,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 768,
"id2label": {
"0": "LABEL_0",
"1": "LABEL_1"
},
"image_size": 224,
"initializer_range": 0.02,
"is_decoder": false,
"is_encoder_decoder": false,
"label2id": {
"LABEL_0": 0,
"LABEL_1": 1
},
"layer_norm_eps": 1e-05,
"length_penalty": 1.0,
"max_length": 20,
"min_length": 0,
"mlp_ratio": 4.0,
"model_type": "swin",
"no_repeat_ngram_size": 0,
"num_beam_groups": 1,
"num_beams": 1,
"num_channels": 3,
"num_heads": [
3,
6,
12,
24
],
"num_layers": 4,
"num_return_sequences": 1,
"out_features": [
"stage1",
"stage2",
"stage3",
"stage4"
],
"out_indices": [
1,
2,
3,
4
],
"output_attentions": false,
"output_hidden_states": false,
"output_scores": false,
"pad_token_id": null,
"patch_size": 4,
"prefix": null,
"problem_type": null,
"pruned_heads": {},
"qkv_bias": true,
"remove_invalid_values": false,
"repetition_penalty": 1.0,
"return_dict": true,
"return_dict_in_generate": false,
"sep_token_id": null,
"stage_names": [
"stem",
"stage1",
"stage2",
"stage3",
"stage4"
],
"suppress_tokens": null,
"task_specific_params": null,
"temperature": 1.0,
"tf_legacy_loss": false,
"tie_encoder_decoder": false,
"tie_word_embeddings": true,
"tokenizer_class": null,
"top_k": 50,
"top_p": 1.0,
"torch_dtype": null,
"torchscript": false,
"transformers_version": "4.31.0",
"typical_p": 1.0,
"use_absolute_embeddings": false,
"use_bfloat16": false,
"window_size": 7
},
"hidden_size": 512,
"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"
},
"initializer_range": 0.02,
"label2id": {
"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,
"cabbage": 91,
"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,
"hanamaki baozi": 62,
"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
},
"loss_ignore_index": 255,
"model_type": "upernet",
"pool_scales": [
1,
2,
3,
6
],
"torch_dtype": "float32",
"transformers_version": null,
"use_auxiliary_head": true
}