| { | |
| "_name_or_path": "facebook/maskformer-swin-base-ade", | |
| "architectures": [ | |
| "MaskFormerForInstanceSegmentation" | |
| ], | |
| "backbone": null, | |
| "backbone_config": { | |
| "depths": [ | |
| 2, | |
| 2, | |
| 18, | |
| 2 | |
| ], | |
| "drop_path_rate": 0.3, | |
| "embed_dim": 128, | |
| "encoder_stride": 32, | |
| "hidden_size": 1024, | |
| "image_size": 384, | |
| "in_channels": 3, | |
| "model_type": "maskformer-swin", | |
| "num_heads": [ | |
| 4, | |
| 8, | |
| 16, | |
| 32 | |
| ], | |
| "out_features": [ | |
| "stage1", | |
| "stage2", | |
| "stage3", | |
| "stage4" | |
| ], | |
| "out_indices": [ | |
| 1, | |
| 2, | |
| 3, | |
| 4 | |
| ], | |
| "path_norm": true, | |
| "pretrain_img_size": 384, | |
| "window_size": 12 | |
| }, | |
| "backbone_kwargs": null, | |
| "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": "unlabeled", | |
| "1": "PV" | |
| }, | |
| "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.43.0.dev0", | |
| "use_auxiliary_loss": false, | |
| "use_pretrained_backbone": false, | |
| "use_timm_backbone": false | |
| } | |