|
{ |
|
"_name_or_path": "facebook/maskformer-swin-base-ade", |
|
"architectures": [ |
|
"MaskFormerForInstanceSegmentation" |
|
], |
|
"backbone_config": { |
|
"depths": [ |
|
2, |
|
2, |
|
18, |
|
2 |
|
], |
|
"drop_path_rate": 0.3, |
|
"embed_dim": 128, |
|
"hidden_size": 1024, |
|
"image_size": 384, |
|
"in_channels": 3, |
|
"model_type": "swin", |
|
"num_heads": [ |
|
4, |
|
8, |
|
16, |
|
32 |
|
], |
|
"out_features": [ |
|
"stage4" |
|
], |
|
"out_indices": [ |
|
4 |
|
], |
|
"path_norm": true, |
|
"pretrain_img_size": 384, |
|
"window_size": 12 |
|
}, |
|
"ce_weight": 1.0, |
|
"cross_entropy_weight": 1.0, |
|
"decoder_config": { |
|
"max_position_embeddings": 1024, |
|
"model_type": "detr", |
|
"scale_embedding": false |
|
}, |
|
"dice_weight": 1.0, |
|
"fpn_feature_size": 256, |
|
"id2label": { |
|
"0": "bg", |
|
"1": "tumor" |
|
}, |
|
"init_std": 0.02, |
|
"init_xavier_std": 1.0, |
|
"label2id": { |
|
"animal": 126, |
|
"arcade machine": 78, |
|
"armchair": 30, |
|
"awning, sunshade, sunblind": 86, |
|
"bag": 115, |
|
"ball": 119, |
|
"bannister, banister, balustrade, balusters, handrail": 95, |
|
"bar": 77, |
|
"barrel, cask": 111, |
|
"base, pedestal, stand": 40, |
|
"basket, handbasket": 112, |
|
"bed": 7, |
|
"bench": 69, |
|
"bicycle": 127, |
|
"blanket, cover": 131, |
|
"blind, screen": 63, |
|
"boat": 76, |
|
"book": 67, |
|
"bookcase": 62, |
|
"booth": 88, |
|
"bottle": 98, |
|
"box": 41, |
|
"bridge, span": 61, |
|
"buffet, counter, sideboard": 99, |
|
"building": 1, |
|
"bulletin board": 144, |
|
"bus": 80, |
|
"cabinet": 10, |
|
"canopy": 106, |
|
"car": 20, |
|
"case, display case, showcase, vitrine": 55, |
|
"ceiling": 5, |
|
"chair": 19, |
|
"chandelier": 85, |
|
"chest of drawers, chest, bureau, dresser": 44, |
|
"clock": 148, |
|
"clothes": 92, |
|
"coffee table": 64, |
|
"column, pillar": 42, |
|
"computer": 74, |
|
"conveyer belt, conveyor belt, conveyer, conveyor, transporter": 105, |
|
"counter": 45, |
|
"countertop": 70, |
|
"cradle": 117, |
|
"crt screen": 141, |
|
"curtain": 18, |
|
"cushion": 39, |
|
"desk": 33, |
|
"dirt track": 91, |
|
"dishwasher": 129, |
|
"door": 14, |
|
"earth, ground": 13, |
|
"escalator, moving staircase, moving stairway": 96, |
|
"falls": 113, |
|
"fan": 139, |
|
"fence": 32, |
|
"field": 29, |
|
"fireplace": 49, |
|
"flag": 149, |
|
"floor": 3, |
|
"flower": 66, |
|
"food, solid food": 120, |
|
"fountain": 104, |
|
"glass, drinking glass": 147, |
|
"grandstand, covered stand": 51, |
|
"grass": 9, |
|
"hill": 68, |
|
"hood, exhaust hood": 133, |
|
"house": 25, |
|
"hovel, hut, hutch, shack, shanty": 79, |
|
"kitchen island": 73, |
|
"lake": 128, |
|
"lamp": 36, |
|
"land, ground, soil": 94, |
|
"light": 82, |
|
"microwave": 124, |
|
"minibike, motorbike": 116, |
|
"mirror": 27, |
|
"monitor": 143, |
|
"mountain, mount": 16, |
|
"ottoman, pouf, pouffe, puff, hassock": 97, |
|
"oven": 118, |
|
"painting, picture": 22, |
|
"palm, palm tree": 72, |
|
"path": 52, |
|
"person": 12, |
|
"pier": 140, |
|
"pillow": 57, |
|
"plane": 90, |
|
"plant": 17, |
|
"plate": 142, |
|
"plaything, toy": 108, |
|
"pole": 93, |
|
"pool": 109, |
|
"pool table, billiard table, snooker table": 56, |
|
"poster, posting, placard, notice, bill, card": 100, |
|
"pot": 125, |
|
"radiator": 146, |
|
"rail": 38, |
|
"refrigerator, icebox": 50, |
|
"river": 60, |
|
"road, route": 6, |
|
"rock, stone": 34, |
|
"rug": 28, |
|
"runway": 54, |
|
"sand": 46, |
|
"sconce": 134, |
|
"screen": 130, |
|
"screen door, screen": 58, |
|
"sculpture": 132, |
|
"sea": 26, |
|
"seat": 31, |
|
"shelf": 24, |
|
"ship": 103, |
|
"shower": 145, |
|
"sidewalk, pavement": 11, |
|
"signboard, sign": 43, |
|
"sink": 47, |
|
"sky": 2, |
|
"skyscraper": 48, |
|
"sofa": 23, |
|
"stage": 101, |
|
"stairs": 53, |
|
"stairway, staircase": 59, |
|
"step, stair": 121, |
|
"stool": 110, |
|
"stove": 71, |
|
"street lamp": 87, |
|
"swivel chair": 75, |
|
"table": 15, |
|
"tank, storage tank": 122, |
|
"tent": 114, |
|
"toilet, can, commode, crapper, pot, potty, stool, throne": 65, |
|
"towel": 81, |
|
"tower": 84, |
|
"trade name": 123, |
|
"traffic light": 136, |
|
"trash can": 138, |
|
"tray": 137, |
|
"tree": 4, |
|
"truck": 83, |
|
"tub": 37, |
|
"tv": 89, |
|
"van": 102, |
|
"vase": 135, |
|
"wall": 0, |
|
"wardrobe, closet, press": 35, |
|
"washer, automatic washer, washing machine": 107, |
|
"water": 21, |
|
"window ": 8 |
|
}, |
|
"mask_feature_size": 256, |
|
"mask_weight": 20.0, |
|
"model_type": "maskformer", |
|
"no_object_weight": 0.1, |
|
"num_attention_heads": 8, |
|
"num_hidden_layers": 6, |
|
"num_queries": 100, |
|
"output_auxiliary_logits": null, |
|
"torch_dtype": "float32", |
|
"transformers_version": "4.34.1", |
|
"use_auxiliary_loss": false |
|
} |
|
|