hf-transformers-bot commited on
Commit
c12ef86
·
verified ·
1 Parent(s): 427fe2f

Update tiny models for HGNetV2Backbone

Browse files
Files changed (3) hide show
  1. config.json +119 -0
  2. model.safetensors +3 -0
  3. preprocessor_config.json +25 -0
config.json ADDED
@@ -0,0 +1,119 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "HGNetV2Backbone"
4
+ ],
5
+ "depths": [
6
+ 1,
7
+ 1,
8
+ 2,
9
+ 1
10
+ ],
11
+ "dtype": "float32",
12
+ "embedding_size": 64,
13
+ "embeddings_size": 10,
14
+ "hidden_act": "relu",
15
+ "hidden_sizes": [
16
+ 64,
17
+ 128,
18
+ 256,
19
+ 512
20
+ ],
21
+ "id2label": {
22
+ "0": "LABEL_0",
23
+ "1": "LABEL_1",
24
+ "2": "LABEL_2"
25
+ },
26
+ "initializer_range": 0.02,
27
+ "label2id": {
28
+ "LABEL_0": 0,
29
+ "LABEL_1": 1,
30
+ "LABEL_2": 2
31
+ },
32
+ "model_type": "hgnet_v2",
33
+ "num_channels": 3,
34
+ "out_features": [
35
+ "stage2",
36
+ "stage3",
37
+ "stage4"
38
+ ],
39
+ "out_indices": [
40
+ 2,
41
+ 3,
42
+ 4
43
+ ],
44
+ "stage_downsample": [
45
+ false,
46
+ true,
47
+ true,
48
+ true
49
+ ],
50
+ "stage_downsample_strides": [
51
+ 2,
52
+ 2,
53
+ 2,
54
+ 2
55
+ ],
56
+ "stage_in_channels": [
57
+ 16,
58
+ 64,
59
+ 128,
60
+ 256
61
+ ],
62
+ "stage_kernel_size": [
63
+ 3,
64
+ 3,
65
+ 5,
66
+ 5
67
+ ],
68
+ "stage_light_block": [
69
+ false,
70
+ false,
71
+ true,
72
+ true
73
+ ],
74
+ "stage_mid_channels": [
75
+ 16,
76
+ 32,
77
+ 64,
78
+ 128
79
+ ],
80
+ "stage_names": [
81
+ "stem",
82
+ "stage1",
83
+ "stage2",
84
+ "stage3",
85
+ "stage4"
86
+ ],
87
+ "stage_num_blocks": [
88
+ 1,
89
+ 1,
90
+ 2,
91
+ 1
92
+ ],
93
+ "stage_numb_of_layers": [
94
+ 3,
95
+ 3,
96
+ 3,
97
+ 3
98
+ ],
99
+ "stage_out_channels": [
100
+ 64,
101
+ 128,
102
+ 256,
103
+ 512
104
+ ],
105
+ "stem_channels": [
106
+ 3,
107
+ 16,
108
+ 16
109
+ ],
110
+ "stem_strides": [
111
+ 2,
112
+ 1,
113
+ 1,
114
+ 2,
115
+ 1
116
+ ],
117
+ "transformers_version": "5.16.0.dev0",
118
+ "use_learnable_affine_block": false
119
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a6432a2b04c858ecb48d2e3fadc00c0965c8e2c33bbe556be60a7b6a477c4c44
3
+ size 2740024
preprocessor_config.json ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_convert_annotations": true,
3
+ "do_normalize": false,
4
+ "do_pad": false,
5
+ "do_rescale": true,
6
+ "do_resize": true,
7
+ "format": "coco_detection",
8
+ "image_mean": [
9
+ 0.485,
10
+ 0.456,
11
+ 0.406
12
+ ],
13
+ "image_processor_type": "RTDetrImageProcessor",
14
+ "image_std": [
15
+ 0.229,
16
+ 0.224,
17
+ 0.225
18
+ ],
19
+ "resample": 2,
20
+ "rescale_factor": 0.00392156862745098,
21
+ "size": {
22
+ "height": 640,
23
+ "width": 640
24
+ }
25
+ }