config.json

{
  "_name_or_path": "nsugianto/detr-resnet50_finetuned_nganimal_primSecv2_lr1e5_dec1e4_bs4",
  "activation_dropout": 0.0,
  "activation_function": "relu",
  "architectures": [
    "DetrForObjectDetection"
  ],
  "attention_dropout": 0.0,
  "auxiliary_loss": false,
  "backbone": "resnet50",
  "backbone_config": null,
  "backbone_kwargs": null,
  "bbox_cost": 5,
  "bbox_loss_coefficient": 5,
  "class_cost": 1,
  "classifier_dropout": 0.0,
  "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,
  "eos_coefficient": 0.1,
  "giou_cost": 2,
  "giou_loss_coefficient": 2,
  "id2label": {
    "0": "animal_batfish",
    "1": "animal_bird",
    "2": "animal_chilomycterusreticulatus",
    "3": "animal_cobia",
    "4": "animal_coral",
    "5": "animal_crownedtoby",
    "6": "animal_cuttlefish",
    "7": "animal_dolphin",
    "8": "animal_fish",
    "9": "animal_greatwhiteshark",
    "10": "animal_greynurseshark",
    "11": "animal_gull",
    "12": "animal_kyphosussectatrix",
    "13": "animal_leopardshark",
    "14": "animal_macrotopus",
    "15": "animal_mantarays",
    "16": "animal_morayeel",
    "17": "animal_oldwife",
    "18": "animal_panulirusversicolor",
    "19": "animal_pinnanobilis",
    "20": "animal_pomfret",
    "21": "animal_puffer",
    "22": "animal_redlionfish",
    "23": "animal_scorpaenascrofa",
    "24": "animal_sealion",
    "25": "animal_seaturtle",
    "26": "animal_starfish",
    "27": "animal_whale",
    "28": "animal_wobbegongcarpetshark",
    "29": "scene_ocean",
    "30": "scene_lighthouse",
    "31": "animal_nudribranch",
    "32": "animal_eaglerays",
    "33": "animal_stringrays",
    "34": "animal_blackrockcod",
    "35": "animal_bleekersdevilfish",
    "36": "animal_estuarycod",
    "37": "animal_humpbackwhales",
    "38": "animal_leatherbackseaturtle",
    "39": "animal_loggerheadseaturtle",
    "40": "animal_greenseaturtle",
    "41": "animal_queenslandgroper"
  },
  "init_std": 0.02,
  "init_xavier_std": 1.0,
  "is_encoder_decoder": true,
  "label2id": {
    "animal_batfish": 0,
    "animal_bird": 1,
    "animal_blackrockcod": 34,
    "animal_bleekersdevilfish": 35,
    "animal_chilomycterusreticulatus": 2,
    "animal_cobia": 3,
    "animal_coral": 4,
    "animal_crownedtoby": 5,
    "animal_cuttlefish": 6,
    "animal_dolphin": 7,
    "animal_eaglerays": 32,
    "animal_estuarycod": 36,
    "animal_fish": 8,
    "animal_greatwhiteshark": 9,
    "animal_greenseaturtle": 40,
    "animal_greynurseshark": 10,
    "animal_gull": 11,
    "animal_humpbackwhales": 37,
    "animal_kyphosussectatrix": 12,
    "animal_leatherbackseaturtle": 38,
    "animal_leopardshark": 13,
    "animal_loggerheadseaturtle": 39,
    "animal_macrotopus": 14,
    "animal_mantarays": 15,
    "animal_morayeel": 16,
    "animal_nudribranch": 31,
    "animal_oldwife": 17,
    "animal_panulirusversicolor": 18,
    "animal_pinnanobilis": 19,
    "animal_pomfret": 20,
    "animal_puffer": 21,
    "animal_queenslandgroper": 41,
    "animal_redlionfish": 22,
    "animal_scorpaenascrofa": 23,
    "animal_sealion": 24,
    "animal_seaturtle": 25,
    "animal_starfish": 26,
    "animal_stringrays": 33,
    "animal_whale": 27,
    "animal_wobbegongcarpetshark": 28,
    "scene_lighthouse": 30,
    "scene_ocean": 29
  },
  "mask_loss_coefficient": 1,
  "max_position_embeddings": 1024,
  "model_type": "detr",
  "num_channels": 3,
  "num_hidden_layers": 6,
  "num_queries": 100,
  "position_embedding_type": "sine",
  "scale_embedding": false,
  "torch_dtype": "float32",
  "transformers_version": "4.41.0.dev0",
  "use_pretrained_backbone": true,
  "use_timm_backbone": true
}