HichTala commited on
Commit
6ca5d6b
·
verified ·
1 Parent(s): cb48226

Upload config

Browse files
Files changed (1) hide show
  1. config.json +82 -2
config.json CHANGED
@@ -1,4 +1,84 @@
1
  {
2
- "_class_name": "DiffusionDet",
3
- "_diffusers_version": "0.32.0.dev0"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
  }
 
1
  {
2
+ "activation": "relu",
3
+ "alpha": 0.25,
4
+ "backbone": "resnet50",
5
+ "backbone_config": null,
6
+ "backbone_kwargs": {
7
+ "in_chans": 3,
8
+ "out_indices": [
9
+ 1,
10
+ 2,
11
+ 3,
12
+ 4
13
+ ]
14
+ },
15
+ "backbone_multiplier": 1.0,
16
+ "class_weight": 2.0,
17
+ "deep_supervision": true,
18
+ "dilation": false,
19
+ "dim_dynamic": 64,
20
+ "dim_feedforward": 2048,
21
+ "dropout": 0.0,
22
+ "fpn_out_channels": 256,
23
+ "gamma": 2.0,
24
+ "giou_weight": 2.0,
25
+ "hidden_dim": 256,
26
+ "l1_weight": 5.0,
27
+ "no_object_weight": 0.1,
28
+ "num_attn_heads": 8,
29
+ "num_channels": 3,
30
+ "num_cls": 1,
31
+ "num_dynamic": 2,
32
+ "num_heads": 6,
33
+ "num_proposals": 300,
34
+ "num_reg": 3,
35
+ "optimizer": "ADAMW",
36
+ "ota_k": 5,
37
+ "pixel_mean": [
38
+ 123.675,
39
+ 116.28,
40
+ 103.53
41
+ ],
42
+ "pixel_std": [
43
+ 58.395,
44
+ 57.12,
45
+ 57.375
46
+ ],
47
+ "pooler_resolution": 7,
48
+ "prior_prob": 0.01,
49
+ "resnet_in_features": [
50
+ "res2",
51
+ "res3",
52
+ "res4",
53
+ "res5"
54
+ ],
55
+ "resnet_out_features": [
56
+ "res2",
57
+ "res3",
58
+ "res4",
59
+ "res5"
60
+ ],
61
+ "roi_head_in_features": [
62
+ "p2",
63
+ "p3",
64
+ "p4",
65
+ "p5"
66
+ ],
67
+ "sample_step": 1,
68
+ "sampling_ratio": 2,
69
+ "snr_scale": 2.0,
70
+ "swin_out_features": [
71
+ 0,
72
+ 1,
73
+ 2,
74
+ 3
75
+ ],
76
+ "swin_size": "B",
77
+ "transformers_version": "4.47.0.dev0",
78
+ "use_fed_loss": false,
79
+ "use_focal": true,
80
+ "use_nms": true,
81
+ "use_pretrained_backbone": true,
82
+ "use_swin_checkpoint": false,
83
+ "use_timm_backbone": true
84
  }