IMvision12 commited on
Commit
5609221
·
verified ·
1 Parent(s): c5da506

Upload kf_config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. kf_config.json +66 -50
kf_config.json CHANGED
@@ -5,54 +5,70 @@
5
  "model_class": "DFineDetect",
6
  "variant": "dfine-large",
7
  "weights": "model.weights.h5",
8
- "config": {
9
- "stem_channels": [
10
- 3,
11
- 32,
12
- 48
13
- ],
14
- "stage_in_channels": [
15
- 48,
16
- 128,
17
- 512,
18
- 1024
19
- ],
20
- "stage_mid_channels": [
21
- 48,
22
- 96,
23
- 192,
24
- 384
25
- ],
26
- "stage_out_channels": [
27
- 128,
28
- 512,
29
- 1024,
30
- 2048
31
- ],
32
- "stage_num_blocks": [
33
- 1,
34
- 1,
35
- 3,
36
- 1
37
- ],
38
- "stage_numb_of_layers": [
39
- 6,
40
- 6,
41
- 6,
42
- 6
43
- ],
44
- "use_lab": false,
45
- "encoder_in_channels": [
46
- 512,
47
- 1024,
48
- 2048
49
- ],
50
- "ccfm_num_blocks": 3,
51
- "decoder_num_layers": 6,
52
- "decoder_n_points": [
53
- 3,
54
- 6,
55
- 3
56
- ]
57
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
58
  }
 
5
  "model_class": "DFineDetect",
6
  "variant": "dfine-large",
7
  "weights": "model.weights.h5",
8
+ "model_type": "d_fine",
9
+ "stem_channels": [
10
+ 3,
11
+ 32,
12
+ 48
13
+ ],
14
+ "stage_in_channels": [
15
+ 48,
16
+ 128,
17
+ 512,
18
+ 1024
19
+ ],
20
+ "stage_mid_channels": [
21
+ 48,
22
+ 96,
23
+ 192,
24
+ 384
25
+ ],
26
+ "stage_out_channels": [
27
+ 128,
28
+ 512,
29
+ 1024,
30
+ 2048
31
+ ],
32
+ "stage_num_blocks": [
33
+ 1,
34
+ 1,
35
+ 3,
36
+ 1
37
+ ],
38
+ "stage_numb_of_layers": [
39
+ 6,
40
+ 6,
41
+ 6,
42
+ 6
43
+ ],
44
+ "use_lab": false,
45
+ "encoder_in_channels": [
46
+ 512,
47
+ 1024,
48
+ 2048
49
+ ],
50
+ "encoder_hidden_dim": 256,
51
+ "encoder_ffn_dim": 1024,
52
+ "encode_proj_layers": [
53
+ 2
54
+ ],
55
+ "hidden_expansion": 1.0,
56
+ "ccfm_num_blocks": 3,
57
+ "hidden_dim": 256,
58
+ "decoder_num_layers": 6,
59
+ "decoder_ffn_dim": 1024,
60
+ "decoder_n_points": [
61
+ 3,
62
+ 6,
63
+ 3
64
+ ],
65
+ "num_feature_levels": 3,
66
+ "feat_strides": [
67
+ 8,
68
+ 16,
69
+ 32
70
+ ],
71
+ "num_classes": 80,
72
+ "num_queries": 300,
73
+ "image_size": 640
74
  }