Hugging Face's logo

zeromodels
/
dfine-large

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

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-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
  }
 
1
  {
2
  "library_name": "kerasformers",
3
+ "kerasformers_version": "1.2.1",
4
  "model_module": "kerasformers.models.dfine",
5
  "model_class": "DFineDetect",
6
  "variant": "dfine-large",
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
+ 32,
15
+ 48
16
+ ],
17
+ "stage_in_channels": [
18
+ 48,
19
+ 128,
20
+ 512,
21
+ 1024
22
+ ],
23
+ "stage_mid_channels": [
24
+ 48,
25
+ 96,
26
+ 192,
27
+ 384
28
+ ],
29
+ "stage_out_channels": [
30
+ 128,
31
+ 512,
32
+ 1024,
33
+ 2048
34
+ ],
35
+ "stage_num_blocks": [
36
+ 1,
37
+ 1,
38
+ 3,
39
+ 1
40
+ ],
41
+ "stage_numb_of_layers": [
42
+ 6,
43
+ 6,
44
+ 6,
45
+ 6
46
+ ],
47
+ "use_lab": false,
48
+ "encoder_in_channels": [
49
+ 512,
50
+ 1024,
51
+ 2048
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": 3,
60
+ "hidden_dim": 256,
61
+ "decoder_num_layers": 6,
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
  }