stevenbucaille commited on
Commit
31bc172
·
verified ·
1 Parent(s): 35c1317

Add rf-detr-medium model

Browse files
Files changed (1) hide show
  1. config.json +2 -5
config.json CHANGED
@@ -20,11 +20,10 @@
20
  "layer_norm_eps": 1e-06,
21
  "layerscale_value": 1.0,
22
  "mlp_ratio": 4,
23
- "model_type": "rf_detr_dinov2_with_registers",
24
  "num_attention_heads": 6,
25
  "num_channels": 3,
26
  "num_hidden_layers": 12,
27
- "num_register_tokens": 0,
28
  "num_windows": 2,
29
  "out_features": [
30
  "stage3",
@@ -56,6 +55,7 @@
56
  "stage11",
57
  "stage12"
58
  ],
 
59
  "use_swiglu_ffn": false,
60
  "window_block_indexes": [
61
  0,
@@ -72,7 +72,6 @@
72
  "backbone_kwargs": null,
73
  "bbox_cost": 5,
74
  "bbox_loss_coefficient": 5,
75
- "bbox_reparam": true,
76
  "class_cost": 2,
77
  "d_model": 256,
78
  "decoder_activation_function": "relu",
@@ -282,7 +281,6 @@
282
  "model_type": "rf_detr",
283
  "num_feature_levels": 1,
284
  "num_queries": 300,
285
- "position_embedding_type": "sine",
286
  "projector_in_channels": [
287
  256
288
  ],
@@ -291,7 +289,6 @@
291
  1.0
292
  ],
293
  "transformers_version": "5.0.0.dev0",
294
- "two_stage": true,
295
  "use_pretrained_backbone": false,
296
  "use_timm_backbone": false
297
  }
 
20
  "layer_norm_eps": 1e-06,
21
  "layerscale_value": 1.0,
22
  "mlp_ratio": 4,
23
+ "model_type": "rf_detr_dinov2",
24
  "num_attention_heads": 6,
25
  "num_channels": 3,
26
  "num_hidden_layers": 12,
 
27
  "num_windows": 2,
28
  "out_features": [
29
  "stage3",
 
55
  "stage11",
56
  "stage12"
57
  ],
58
+ "use_mask_token": true,
59
  "use_swiglu_ffn": false,
60
  "window_block_indexes": [
61
  0,
 
72
  "backbone_kwargs": null,
73
  "bbox_cost": 5,
74
  "bbox_loss_coefficient": 5,
 
75
  "class_cost": 2,
76
  "d_model": 256,
77
  "decoder_activation_function": "relu",
 
281
  "model_type": "rf_detr",
282
  "num_feature_levels": 1,
283
  "num_queries": 300,
 
284
  "projector_in_channels": [
285
  256
286
  ],
 
289
  1.0
290
  ],
291
  "transformers_version": "5.0.0.dev0",
 
292
  "use_pretrained_backbone": false,
293
  "use_timm_backbone": false
294
  }