config.json

{
  "_name_or_path": "/content/drive/MyDrive/Airbus/checkpoints/model_0.8054",
  "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": "background",
    "1": "ship"
  },
  "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.33.2",
  "use_auxiliary_loss": false
}