Add rf-detr-medium model
Browse files- 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": "
|
| 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 |
}
|