Add rf-detr-medium model
Browse files- config.json +4 -1
- model.safetensors +2 -2
config.json
CHANGED
|
@@ -72,6 +72,7 @@
|
|
| 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,6 +282,7 @@
|
|
| 281 |
"model_type": "rf_detr",
|
| 282 |
"num_feature_levels": 1,
|
| 283 |
"num_queries": 300,
|
|
|
|
| 284 |
"projector_in_channels": [
|
| 285 |
256
|
| 286 |
],
|
|
@@ -288,7 +290,8 @@
|
|
| 288 |
"projector_scale_factors": [
|
| 289 |
1.0
|
| 290 |
],
|
| 291 |
-
"transformers_version": "
|
|
|
|
| 292 |
"use_pretrained_backbone": false,
|
| 293 |
"use_timm_backbone": false
|
| 294 |
}
|
|
|
|
| 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 |
"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 |
],
|
|
|
|
| 290 |
"projector_scale_factors": [
|
| 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 |
}
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2820745fc3ba25b86d2d85afef9dfd992f9522e9bd3112c5af9a353ca22e966a
|
| 3 |
+
size 134812480
|