config.json

{
  "_name_or_path": "pretrain/checkpoint-15500",
  "activation_dropout": 0.0,
  "activation_function": "relu",
  "architectures": [
    "ConditionalDetrForObjectDetection"
  ],
  "attention_dropout": 0.0,
  "auxiliary_loss": false,
  "backbone": "resnet50",
  "backbone_config": null,
  "bbox_cost": 5,
  "bbox_loss_coefficient": 5,
  "class_cost": 2,
  "cls_loss_coefficient": 2,
  "d_model": 256,
  "decoder_attention_heads": 8,
  "decoder_ffn_dim": 2048,
  "decoder_layerdrop": 0.0,
  "decoder_layers": 6,
  "dice_loss_coefficient": 1,
  "dilation": false,
  "dropout": 0.1,
  "encoder_attention_heads": 8,
  "encoder_ffn_dim": 2048,
  "encoder_layerdrop": 0.0,
  "encoder_layers": 6,
  "focal_alpha": 0.25,
  "giou_cost": 2,
  "giou_loss_coefficient": 2,
  "id2label": {
    "0": "no-object",
    "1": "Kellogg's Corn Flakes",
    "2": "Honey Stars",
    "3": "Koko Krunch",
    "4": "Fita",
    "5": "Mamon",
    "6": "Buttercake",
    "7": "Wafer",
    "8": "Choco Crunchies",
    "9": "iHop",
    "10": "Butterlicious",
    "11": "Pik-nik",
    "12": "Pringles",
    "13": "Super Fruits",
    "14": "Locally Mango",
    "15": "Camomille",
    "16": "Oishi Prawn",
    "17": "Kimchi Noodles",
    "18": "Grahams",
    "19": "Hansel",
    "20": "Danisa",
    "21": "Chocolate Chips",
    "22": "Fresh Milk",
    "23": "Bear brand",
    "24": "Chuckie",
    "25": "Head"
  },
  "init_std": 0.02,
  "init_xavier_std": 1.0,
  "is_encoder_decoder": true,
  "label2id": {
    "Bear brand": "23",
    "Buttercake": "6",
    "Butterlicious": "10",
    "Camomille": "15",
    "Choco Crunchies": "8",
    "Chocolate Chips": "21",
    "Chuckie": "24",
    "Danisa": "20",
    "Fita": "4",
    "Fresh Milk": "22",
    "Grahams": "18",
    "Hansel": "19",
    "Head": "25",
    "Honey Stars": "2",
    "Kellogg's Corn Flakes": "1",
    "Kimchi Noodles": "17",
    "Koko Krunch": "3",
    "Locally Mango": "14",
    "Mamon": "5",
    "Oishi Prawn": "16",
    "Pik-nik": "11",
    "Pringles": "12",
    "Super Fruits": "13",
    "Wafer": "7",
    "iHop": "9",
    "no-object": "0"
  },
  "mask_loss_coefficient": 1,
  "max_position_embeddings": 1024,
  "model_type": "conditional_detr",
  "num_channels": 3,
  "num_hidden_layers": 6,
  "num_queries": 300,
  "position_embedding_type": "sine",
  "scale_embedding": false,
  "torch_dtype": "float32",
  "transformers_version": "4.28.0",
  "use_pretrained_backbone": true,
  "use_timm_backbone": true
}