Add config file at 2025-10-05 07:06:56

config.json

@@ -20,6 +20,7 @@
   "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,
@@ -28,7 +29,6 @@
   "dice_loss_coefficient": 1,
   "dilation": false,
   "dropout": 0.1,
-  "dtype": "float32",
   "encoder_attention_heads": 8,
   "encoder_ffn_dim": 2048,
   "encoder_layerdrop": 0.0,
@@ -37,37 +37,39 @@
   "giou_cost": 2,
   "giou_loss_coefficient": 2,
   "id2label": {
-    "0": "LABEL_0",
-    "1": "LABEL_1",
-    "2": "LABEL_2",
-    "3": "LABEL_3",
-    "4": "LABEL_4",
-    "5": "LABEL_5",
-    "6": "LABEL_6",
-    "7": "LABEL_7",
-    "8": "LABEL_8"
+    "0": "Trout",
+    "1": "Shrimp",
+    "2": "Gilt-Head Bream",
+    "3": "Hourse Mackerel",
+    "4": "Red Mullet",
+    "5": "Striped Red Mullet",
+    "6": "Red Sea Bream",
+    "7": "Black Sea Sprat",
+    "8": "Sea Bass"
   },
   "init_std": 0.02,
   "init_xavier_std": 1.0,
   "is_encoder_decoder": true,
   "label2id": {
-    "LABEL_0": 0,
-    "LABEL_1": 1,
-    "LABEL_2": 2,
-    "LABEL_3": 3,
-    "LABEL_4": 4,
-    "LABEL_5": 5,
-    "LABEL_6": 6,
-    "LABEL_7": 7,
-    "LABEL_8": 8
+    "Black Sea Sprat": 7,
+    "Gilt-Head Bream": 2,
+    "Hourse Mackerel": 3,
+    "Red Mullet": 4,
+    "Red Sea Bream": 6,
+    "Sea Bass": 8,
+    "Shrimp": 1,
+    "Striped Red Mullet": 5,
+    "Trout": 0
   },
   "mask_loss_coefficient": 1,
+  "max_position_embeddings": 1024,
   "model_name": "Fish_Segmentation_Model_Fine_Tuning_DETR",
   "model_type": "detr",
   "num_channels": 3,
   "num_hidden_layers": 6,
   "num_queries": 100,
   "position_embedding_type": "sine",
+  "scale_embedding": false,
   "transformers_version": "4.56.2",
   "use_pretrained_backbone": true,
   "use_timm_backbone": true