Add config from convert_rt_detr_original_pytorch_checkpoint_to_pytorch.py

config.json

@@ -10,16 +10,6 @@
   ],
   "attention_dropout": 0.0,
   "auxiliary_loss": true,
-  "backbone": "resnet34d",
-  "backbone_config": null,
-  "backbone_kwargs": {
-    "features_only": true,
-    "out_indices": [
-      2,
-      3,
-      4
-    ]
-  },
   "batch_norm_eps": 1e-05,
   "box_noise_scale": 1.0,
   "d_model": 256,
@@ -33,8 +23,17 @@
   ],
   "decoder_layers": 4,
   "decoder_n_points": 4,
+  "depths": [
+    3,
+    4,
+    6,
+    3
+  ],
   "disable_custom_kernels": true,
+  "downsample_in_bottleneck": false,
+  "downsample_in_first_stage": false,
   "dropout": 0.0,
+  "embedding_size": 64,
   "encode_proj_layers": [
     2
   ],
@@ -57,7 +56,14 @@
   ],
   "focal_loss_alpha": 0.75,
   "focal_loss_gamma": 2.0,
+  "hidden_act": "relu",
   "hidden_expansion": 0.5,
+  "hidden_sizes": [
+    64,
+    128,
+    256,
+    512
+  ],
   "id2label": {
     "0": "person",
     "1": "bicycle",
@@ -226,6 +232,7 @@
   },
   "label_noise_ratio": 0.5,
   "layer_norm_eps": 1e-05,
+  "layer_type": "basic",
   "learn_initial_query": false,
   "matcher_alpha": 0.25,
   "matcher_bbox_cost": 5.0,
@@ -234,15 +241,31 @@
   "matcher_giou_cost": 2.0,
   "model_type": "rt_detr",
   "normalize_before": false,
+  "num_channels": 3,
   "num_denoising": 100,
   "num_feature_levels": 3,
   "num_queries": 300,
+  "out_features": [
+    "stage2",
+    "stage3",
+    "stage4"
+  ],
+  "out_indices": [
+    2,
+    3,
+    4
+  ],
   "positional_encoding_temperature": 10000,
+  "stage_names": [
+    "stem",
+    "stage1",
+    "stage2",
+    "stage3",
+    "stage4"
+  ],
   "torch_dtype": "float32",
   "transformers_version": "4.42.0.dev0",
   "use_focal_loss": true,
-  "use_pretrained_backbone": true,
-  "use_timm_backbone": true,
   "weight_loss_bbox": 5.0,
   "weight_loss_giou": 2.0,
   "weight_loss_vfl": 1.0,