Hugging Face's logo

zeromodels
/
dfine-medium

Object Detection
Keras
PyTorch
JAX
TensorFlow
zeromodels
d-fine
dfine
detr
Model card Files
xet
 Community
IMvision12 commited on
Commit
d629251
·
verified ·
1 Parent(s): 680647b

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-medium",
7
  "weights": "model.weights.h5",
8
- "config": {
9
- "stem_channels": [
10
- 3,
11
- 24,
12
- 32
13
- ],
14
- "stage_in_channels": [
15
- 32,
16
- 96,
17
- 384,
18
- 768
19
- ],
20
- "stage_mid_channels": [
21
- 32,
22
- 64,
23
- 128,
24
- 256
25
- ],
26
- "stage_out_channels": [
27
- 96,
28
- 384,
29
- 768,
30
- 1536
31
- ],
32
- "stage_num_blocks": [
33
- 1,
34
- 1,
35
- 3,
36
- 1
37
- ],
38
- "stage_numb_of_layers": [
39
- 4,
40
- 4,
41
- 4,
42
- 4
43
- ],
44
- "use_lab": true,
45
- "encoder_in_channels": [
46
- 384,
47
- 768,
48
- 1536
49
- ],
50
- "ccfm_num_blocks": 2,
51
- "decoder_num_layers": 4,
52
- "decoder_n_points": [
53
- 3,
54
- 6,
55
- 3
56
- ]
57
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
58
  }
 
5
  "model_class": "DFineDetect",
6
  "variant": "dfine-medium",
7
  "weights": "model.weights.h5",
8
+ "model_type": "d_fine",
9
+ "stem_channels": [
10
+ 3,
11
+ 24,
12
+ 32
13
+ ],
14
+ "stage_in_channels": [
15
+ 32,
16
+ 96,
17
+ 384,
18
+ 768
19
+ ],
20
+ "stage_mid_channels": [
21
+ 32,
22
+ 64,
23
+ 128,
24
+ 256
25
+ ],
26
+ "stage_out_channels": [
27
+ 96,
28
+ 384,
29
+ 768,
30
+ 1536
31
+ ],
32
+ "stage_num_blocks": [
33
+ 1,
34
+ 1,
35
+ 3,
36
+ 1
37
+ ],
38
+ "stage_numb_of_layers": [
39
+ 4,
40
+ 4,
41
+ 4,
42
+ 4
43
+ ],
44
+ "use_lab": true,
45
+ "encoder_in_channels": [
46
+ 384,
47
+ 768,
48
+ 1536
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": 2,
57
+ "hidden_dim": 256,
58
+ "decoder_num_layers": 4,
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
  }