Hugging Face's logo

Bekhouche
/
VGG-CTC-STR

Transformers
Safetensors
Model card Files
xet
Community
Bekhouche commited on
Commit
ca1f49e
·
verified ·
1 Parent(s): 4b0646b

Upload Pipeline

Browse files
Files changed (2) hide show
  1. config.json +187 -187
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,187 +1,187 @@
1
- {
2
- "inputs": [
3
- "images"
4
- ],
5
- "modules": {
6
- "avg_pool": {
7
- "config": {
8
- "args": {
9
- "output_size": [
10
- null,
11
- 1
12
- ]
13
- }
14
- },
15
- "type": "torch.nn.AdaptiveAvgPool2d"
16
- },
17
- "feature_extraction": {
18
- "config": {
19
- "args": {
20
- "input_channel": 1,
21
- "output_channel": 512,
22
- "variant": "DTRB"
23
- }
24
- },
25
- "type": "DeepTextRecognition.VGGModel"
26
- },
27
- "permute": {
28
- "config": {
29
- "args": {
30
- "dims": [
31
- 0,
32
- 3,
33
- 1,
34
- 2
35
- ]
36
- }
37
- },
38
- "type": "torch.permute"
39
- },
40
- "prediction": {
41
- "config": {
42
- "args": {
43
- "in_features": 512,
44
- "out_features": 37
45
- }
46
- },
47
- "type": "torch.nn.Linear"
48
- },
49
- "processing": {
50
- "config": {
51
- "args": {
52
- "channels_size": 1,
53
- "image_size": [
54
- 32,
55
- 100
56
- ],
57
- "padding": "left"
58
- }
59
- },
60
- "type": "DeepTextRecognition.ImageProcessor"
61
- },
62
- "squeeze": {
63
- "config": {
64
- "args": {
65
- "dim": 3
66
- }
67
- },
68
- "type": "torch.squeeze"
69
- },
70
- "tokenizer": {
71
- "config": {
72
- "args": {
73
- "characters": [
74
- "0",
75
- "1",
76
- "2",
77
- "3",
78
- "4",
79
- "5",
80
- "6",
81
- "7",
82
- "8",
83
- "9",
84
- "a",
85
- "b",
86
- "c",
87
- "d",
88
- "e",
89
- "f",
90
- "g",
91
- "h",
92
- "i",
93
- "j",
94
- "k",
95
- "l",
96
- "m",
97
- "n",
98
- "o",
99
- "p",
100
- "q",
101
- "r",
102
- "s",
103
- "t",
104
- "u",
105
- "v",
106
- "w",
107
- "x",
108
- "y",
109
- "z"
110
- ],
111
- "max_length": 25
112
- }
113
- },
114
- "type": "DeepTextRecognition.CTCTokenizer"
115
- }
116
- },
117
- "order": [
118
- "processing",
119
- "feature_extraction",
120
- "permute",
121
- "avg_pool",
122
- "squeeze",
123
- "prediction",
124
- "tokenizer"
125
- ],
126
- "outputs": [
127
- "tokenizer:labels"
128
- ],
129
- "routing": {
130
- "avg_pool": {
131
- "inputs": [
132
- "permute:permuted_features"
133
- ],
134
- "outputs": [
135
- "avg_pool:pooled_features"
136
- ]
137
- },
138
- "feature_extraction": {
139
- "inputs": [
140
- "processing:processed_images"
141
- ],
142
- "outputs": [
143
- "feature_extraction:extracted_features"
144
- ]
145
- },
146
- "permute": {
147
- "inputs": [
148
- "feature_extraction:extracted_features"
149
- ],
150
- "outputs": [
151
- "permute:permuted_features"
152
- ]
153
- },
154
- "prediction": {
155
- "inputs": [
156
- "squeeze:squeezed_features"
157
- ],
158
- "outputs": [
159
- "prediction:predictions"
160
- ]
161
- },
162
- "processing": {
163
- "inputs": [
164
- "images"
165
- ],
166
- "outputs": [
167
- "processing:processed_images"
168
- ]
169
- },
170
- "squeeze": {
171
- "inputs": [
172
- "avg_pool:pooled_features"
173
- ],
174
- "outputs": [
175
- "squeeze:squeezed_features"
176
- ]
177
- },
178
- "tokenizer": {
179
- "inputs": [
180
- "prediction:predictions"
181
- ],
182
- "outputs": [
183
- "tokenizer:labels"
184
- ]
185
- }
186
- }
187
- }
 
1
+ {
2
+ "inputs": [
3
+ "images"
4
+ ],
5
+ "modules": {
6
+ "avg_pool": {
7
+ "config": {
8
+ "args": {
9
+ "output_size": [
10
+ null,
11
+ 1
12
+ ]
13
+ }
14
+ },
15
+ "type": "DeepTextRecognition.AdaptiveAvgPoolModule"
16
+ },
17
+ "feature_extraction": {
18
+ "config": {
19
+ "args": {
20
+ "input_channel": 1,
21
+ "output_channel": 512,
22
+ "variant": "DTRB"
23
+ }
24
+ },
25
+ "type": "DeepTextRecognition.VGGModel"
26
+ },
27
+ "permute": {
28
+ "config": {
29
+ "args": {
30
+ "dims": [
31
+ 0,
32
+ 3,
33
+ 1,
34
+ 2
35
+ ]
36
+ }
37
+ },
38
+ "type": "DeepTextRecognition.PermuteModule"
39
+ },
40
+ "prediction": {
41
+ "config": {
42
+ "args": {
43
+ "in_features": 512,
44
+ "out_features": 37
45
+ }
46
+ },
47
+ "type": "torch.nn.Linear"
48
+ },
49
+ "processing": {
50
+ "config": {
51
+ "args": {
52
+ "channels_size": 1,
53
+ "image_size": [
54
+ 32,
55
+ 100
56
+ ],
57
+ "padding": "left"
58
+ }
59
+ },
60
+ "type": "DeepTextRecognition.ImageProcessor"
61
+ },
62
+ "squeeze": {
63
+ "config": {
64
+ "args": {
65
+ "dim": 3
66
+ }
67
+ },
68
+ "type": "DeepTextRecognition.SqueezeModule"
69
+ },
70
+ "tokenizer": {
71
+ "config": {
72
+ "args": {
73
+ "characters": [
74
+ "0",
75
+ "1",
76
+ "2",
77
+ "3",
78
+ "4",
79
+ "5",
80
+ "6",
81
+ "7",
82
+ "8",
83
+ "9",
84
+ "a",
85
+ "b",
86
+ "c",
87
+ "d",
88
+ "e",
89
+ "f",
90
+ "g",
91
+ "h",
92
+ "i",
93
+ "j",
94
+ "k",
95
+ "l",
96
+ "m",
97
+ "n",
98
+ "o",
99
+ "p",
100
+ "q",
101
+ "r",
102
+ "s",
103
+ "t",
104
+ "u",
105
+ "v",
106
+ "w",
107
+ "x",
108
+ "y",
109
+ "z"
110
+ ],
111
+ "max_length": 25
112
+ }
113
+ },
114
+ "type": "DeepTextRecognition.CTCTokenizer"
115
+ }
116
+ },
117
+ "order": [
118
+ "processing",
119
+ "feature_extraction",
120
+ "permute",
121
+ "avg_pool",
122
+ "squeeze",
123
+ "prediction",
124
+ "tokenizer"
125
+ ],
126
+ "outputs": [
127
+ "tokenizer:labels"
128
+ ],
129
+ "routing": {
130
+ "avg_pool": {
131
+ "inputs": [
132
+ "permute:permuted_features"
133
+ ],
134
+ "outputs": [
135
+ "avg_pool:pooled_features"
136
+ ]
137
+ },
138
+ "feature_extraction": {
139
+ "inputs": [
140
+ "processing:processed_images"
141
+ ],
142
+ "outputs": [
143
+ "feature_extraction:extracted_features"
144
+ ]
145
+ },
146
+ "permute": {
147
+ "inputs": [
148
+ "feature_extraction:extracted_features"
149
+ ],
150
+ "outputs": [
151
+ "permute:permuted_features"
152
+ ]
153
+ },
154
+ "prediction": {
155
+ "inputs": [
156
+ "squeeze:squeezed_features"
157
+ ],
158
+ "outputs": [
159
+ "prediction:predictions"
160
+ ]
161
+ },
162
+ "processing": {
163
+ "inputs": [
164
+ "images"
165
+ ],
166
+ "outputs": [
167
+ "processing:processed_images"
168
+ ]
169
+ },
170
+ "squeeze": {
171
+ "inputs": [
172
+ "avg_pool:pooled_features"
173
+ ],
174
+ "outputs": [
175
+ "squeeze:squeezed_features"
176
+ ]
177
+ },
178
+ "tokenizer": {
179
+ "inputs": [
180
+ "prediction:predictions"
181
+ ],
182
+ "outputs": [
183
+ "tokenizer:labels"
184
+ ]
185
+ }
186
+ }
187
+ }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7f1643386fa79650343d9efdbbc76523754ac57c8203fbf659bfe89aa13a5796
3
- size 22286156
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ad2ccf2e101df4d3ec32055986721105715be3a61529b108c076b61716bdce6f
3
+ size 22286348