Commit
·
f30cbbc
1
Parent(s):
84cde37
Add config from convert_rt_detr_original_pytorch_checkpoint_to_pytorch.py
Browse files- config.json +3 -8
config.json
CHANGED
@@ -1,8 +1,5 @@
|
|
1 |
{
|
2 |
"act_encoder": "silu",
|
3 |
-
"architectures": [
|
4 |
-
"RTDetrModel"
|
5 |
-
],
|
6 |
"backbone_config": {
|
7 |
"backbone": "resnet50d",
|
8 |
"freeze_batch_norm_2d": true,
|
@@ -17,6 +14,9 @@
|
|
17 |
"box_noise_scale": 1.0,
|
18 |
"dim_feedforward": 1024,
|
19 |
"dropout": 0.0,
|
|
|
|
|
|
|
20 |
"encoder_attention_heads": 8,
|
21 |
"eos_coefficient": 0.1,
|
22 |
"eval_idx": -1,
|
@@ -217,7 +217,6 @@
|
|
217 |
"matcher_giou_cost": 2.0,
|
218 |
"model_type": "rt_detr",
|
219 |
"normalize_before": false,
|
220 |
-
"num_classes": 80,
|
221 |
"num_decoder_layers": 6,
|
222 |
"num_decoder_points": 4,
|
223 |
"num_denoising": 100,
|
@@ -225,12 +224,8 @@
|
|
225 |
"num_levels": 3,
|
226 |
"num_queries": 300,
|
227 |
"pe_temperature": 10000,
|
228 |
-
"torch_dtype": "float32",
|
229 |
"transformers_version": "4.36.0.dev0",
|
230 |
"use_aux_loss": true,
|
231 |
-
"use_encoder_idx": [
|
232 |
-
2
|
233 |
-
],
|
234 |
"use_focal_loss": true,
|
235 |
"weight_loss_bbox": 5.0,
|
236 |
"weight_loss_giou": 2.0,
|
|
|
1 |
{
|
2 |
"act_encoder": "silu",
|
|
|
|
|
|
|
3 |
"backbone_config": {
|
4 |
"backbone": "resnet50d",
|
5 |
"freeze_batch_norm_2d": true,
|
|
|
14 |
"box_noise_scale": 1.0,
|
15 |
"dim_feedforward": 1024,
|
16 |
"dropout": 0.0,
|
17 |
+
"encode_proj_layers": [
|
18 |
+
2
|
19 |
+
],
|
20 |
"encoder_attention_heads": 8,
|
21 |
"eos_coefficient": 0.1,
|
22 |
"eval_idx": -1,
|
|
|
217 |
"matcher_giou_cost": 2.0,
|
218 |
"model_type": "rt_detr",
|
219 |
"normalize_before": false,
|
|
|
220 |
"num_decoder_layers": 6,
|
221 |
"num_decoder_points": 4,
|
222 |
"num_denoising": 100,
|
|
|
224 |
"num_levels": 3,
|
225 |
"num_queries": 300,
|
226 |
"pe_temperature": 10000,
|
|
|
227 |
"transformers_version": "4.36.0.dev0",
|
228 |
"use_aux_loss": true,
|
|
|
|
|
|
|
229 |
"use_focal_loss": true,
|
230 |
"weight_loss_bbox": 5.0,
|
231 |
"weight_loss_giou": 2.0,
|