Add doctr-dummy-tf-vit-b-v2 model
Browse files- README.md +39 -0
- config.json +149 -0
- tf_model.weights.h5 +3 -0
README.md
ADDED
@@ -0,0 +1,39 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
|
2 |
+
---
|
3 |
+
language: en
|
4 |
+
---
|
5 |
+
|
6 |
+
<p align="center">
|
7 |
+
<img src="https://doctr-static.mindee.com/models?id=v0.3.1/Logo_doctr.gif&src=0" width="60%">
|
8 |
+
</p>
|
9 |
+
|
10 |
+
**Optical Character Recognition made seamless & accessible to anyone, powered by TensorFlow 2 & PyTorch**
|
11 |
+
|
12 |
+
## Task: classification
|
13 |
+
|
14 |
+
https://github.com/mindee/doctr
|
15 |
+
|
16 |
+
### Example usage:
|
17 |
+
|
18 |
+
```python
|
19 |
+
>>> from doctr.io import DocumentFile
|
20 |
+
>>> from doctr.models import ocr_predictor, from_hub
|
21 |
+
|
22 |
+
>>> img = DocumentFile.from_images(['<image_path>'])
|
23 |
+
>>> # Load your model from the hub
|
24 |
+
>>> model = from_hub('mindee/my-model')
|
25 |
+
|
26 |
+
>>> # Pass it to the predictor
|
27 |
+
>>> # If your model is a recognition model:
|
28 |
+
>>> predictor = ocr_predictor(det_arch='db_mobilenet_v3_large',
|
29 |
+
>>> reco_arch=model,
|
30 |
+
>>> pretrained=True)
|
31 |
+
|
32 |
+
>>> # If your model is a detection model:
|
33 |
+
>>> predictor = ocr_predictor(det_arch=model,
|
34 |
+
>>> reco_arch='crnn_mobilenet_v3_small',
|
35 |
+
>>> pretrained=True)
|
36 |
+
|
37 |
+
>>> # Get your predictions
|
38 |
+
>>> res = predictor(img)
|
39 |
+
```
|
config.json
ADDED
@@ -0,0 +1,149 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"mean": [
|
3 |
+
0.694,
|
4 |
+
0.695,
|
5 |
+
0.693
|
6 |
+
],
|
7 |
+
"std": [
|
8 |
+
0.299,
|
9 |
+
0.296,
|
10 |
+
0.301
|
11 |
+
],
|
12 |
+
"input_shape": [
|
13 |
+
32,
|
14 |
+
32,
|
15 |
+
3
|
16 |
+
],
|
17 |
+
"classes": [
|
18 |
+
"0",
|
19 |
+
"1",
|
20 |
+
"2",
|
21 |
+
"3",
|
22 |
+
"4",
|
23 |
+
"5",
|
24 |
+
"6",
|
25 |
+
"7",
|
26 |
+
"8",
|
27 |
+
"9",
|
28 |
+
"a",
|
29 |
+
"b",
|
30 |
+
"c",
|
31 |
+
"d",
|
32 |
+
"e",
|
33 |
+
"f",
|
34 |
+
"g",
|
35 |
+
"h",
|
36 |
+
"i",
|
37 |
+
"j",
|
38 |
+
"k",
|
39 |
+
"l",
|
40 |
+
"m",
|
41 |
+
"n",
|
42 |
+
"o",
|
43 |
+
"p",
|
44 |
+
"q",
|
45 |
+
"r",
|
46 |
+
"s",
|
47 |
+
"t",
|
48 |
+
"u",
|
49 |
+
"v",
|
50 |
+
"w",
|
51 |
+
"x",
|
52 |
+
"y",
|
53 |
+
"z",
|
54 |
+
"A",
|
55 |
+
"B",
|
56 |
+
"C",
|
57 |
+
"D",
|
58 |
+
"E",
|
59 |
+
"F",
|
60 |
+
"G",
|
61 |
+
"H",
|
62 |
+
"I",
|
63 |
+
"J",
|
64 |
+
"K",
|
65 |
+
"L",
|
66 |
+
"M",
|
67 |
+
"N",
|
68 |
+
"O",
|
69 |
+
"P",
|
70 |
+
"Q",
|
71 |
+
"R",
|
72 |
+
"S",
|
73 |
+
"T",
|
74 |
+
"U",
|
75 |
+
"V",
|
76 |
+
"W",
|
77 |
+
"X",
|
78 |
+
"Y",
|
79 |
+
"Z",
|
80 |
+
"!",
|
81 |
+
"\"",
|
82 |
+
"#",
|
83 |
+
"$",
|
84 |
+
"%",
|
85 |
+
"&",
|
86 |
+
"'",
|
87 |
+
"(",
|
88 |
+
")",
|
89 |
+
"*",
|
90 |
+
"+",
|
91 |
+
",",
|
92 |
+
"-",
|
93 |
+
".",
|
94 |
+
"/",
|
95 |
+
":",
|
96 |
+
";",
|
97 |
+
"<",
|
98 |
+
"=",
|
99 |
+
">",
|
100 |
+
"?",
|
101 |
+
"@",
|
102 |
+
"[",
|
103 |
+
"\\",
|
104 |
+
"]",
|
105 |
+
"^",
|
106 |
+
"_",
|
107 |
+
"`",
|
108 |
+
"{",
|
109 |
+
"|",
|
110 |
+
"}",
|
111 |
+
"~",
|
112 |
+
"°",
|
113 |
+
"£",
|
114 |
+
"€",
|
115 |
+
"¥",
|
116 |
+
"¢",
|
117 |
+
"฿",
|
118 |
+
"à",
|
119 |
+
"â",
|
120 |
+
"é",
|
121 |
+
"è",
|
122 |
+
"ê",
|
123 |
+
"ë",
|
124 |
+
"î",
|
125 |
+
"ï",
|
126 |
+
"ô",
|
127 |
+
"ù",
|
128 |
+
"û",
|
129 |
+
"ü",
|
130 |
+
"ç",
|
131 |
+
"À",
|
132 |
+
"Â",
|
133 |
+
"É",
|
134 |
+
"È",
|
135 |
+
"Ê",
|
136 |
+
"Ë",
|
137 |
+
"Î",
|
138 |
+
"Ï",
|
139 |
+
"Ô",
|
140 |
+
"Ù",
|
141 |
+
"Û",
|
142 |
+
"Ü",
|
143 |
+
"Ç"
|
144 |
+
],
|
145 |
+
"url": "https://github.com/mindee/doctr/releases/download/v0.9.0/vit_b-c64705bd.weights.h5",
|
146 |
+
"num_classes": 126,
|
147 |
+
"arch": "vit_b",
|
148 |
+
"task": "classification"
|
149 |
+
}
|
tf_model.weights.h5
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c64705bd0bcd747747cf6d277a0858f3eb2f5bf0d7a236e0f920348aeafb8e1e
|
3 |
+
size 341123848
|