Hugging Face's logo

zeromodels
/
dfine-medium

Object Detection
KerasFormers
Keras
PyTorch
JAX
TensorFlow
d-fine
dfine
detr
Model card Files
xet
 Community
IMvision12 commited on
Commit
1aea378
·
verified ·
1 Parent(s): 4abee9b

Upgrade kf_config.json to v2 (nested config + dtype)

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