onnx/config.json
1.3 KB · 62 lines · json Raw
1 {
2 "_attn_implementation_autoset": true,
3 "activation_dropout": 0.0,
4 "activation_function": "relu",
5 "architectures": [
6 "DetrForObjectDetection"
7 ],
8 "attention_dropout": 0.0,
9 "auxiliary_loss": false,
10 "backbone": "resnet50",
11 "backbone_config": null,
12 "backbone_kwargs": {
13 "in_chans": 3,
14 "out_indices": [
15 1,
16 2,
17 3,
18 4
19 ]
20 },
21 "bbox_cost": 5,
22 "bbox_loss_coefficient": 5,
23 "class_cost": 1,
24 "d_model": 256,
25 "decoder_attention_heads": 8,
26 "decoder_ffn_dim": 2048,
27 "decoder_layerdrop": 0.0,
28 "decoder_layers": 6,
29 "dice_loss_coefficient": 1,
30 "dilation": false,
31 "dropout": 0.1,
32 "encoder_attention_heads": 8,
33 "encoder_ffn_dim": 2048,
34 "encoder_layerdrop": 0.0,
35 "encoder_layers": 6,
36 "eos_coefficient": 0.1,
37 "giou_cost": 2,
38 "giou_loss_coefficient": 2,
39 "id2label": {
40 "0": "table",
41 "1": "table"
42 },
43 "init_std": 0.02,
44 "init_xavier_std": 1.0,
45 "is_encoder_decoder": true,
46 "label2id": {
47 "table": 0
48 },
49 "mask_loss_coefficient": 1,
50 "max_position_embeddings": 1024,
51 "model_type": "detr",
52 "num_channels": 3,
53 "num_hidden_layers": 6,
54 "num_queries": 100,
55 "position_embedding_type": "sine",
56 "scale_embedding": false,
57 "torch_dtype": "float32",
58 "transformers_version": "4.51.3",
59 "use_pretrained_backbone": true,
60 "use_timm_backbone": true
61 }
62