Felix92 commited on
Commit
f15d8e7
1 Parent(s): 572115a

Add onnxtr-crnn-vgg16-bn model

Browse files
Files changed (3) hide show
  1. README.md +37 -0
  2. config.json +22 -0
  3. model.onnx +3 -0
README.md ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ ---
3
+ language: en
4
+ ---
5
+
6
+ <p align="center">
7
+ <img src="https://github.com/felixdittrich92/OnnxTR/raw/main/docs/images/logo.jpg" width="40%">
8
+ </p>
9
+
10
+ **Optical Character Recognition made seamless & accessible to anyone, powered by Onnxruntime**
11
+
12
+ ## Task: recognition
13
+
14
+ https://github.com/felixdittrich92/OnnxTR
15
+
16
+ ### Example usage:
17
+
18
+ ```python
19
+ >>> from onnxtr.io import DocumentFile
20
+ >>> from onnxtr.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('onnxtr/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
+
31
+ >>> # If your model is a detection model:
32
+ >>> predictor = ocr_predictor(det_arch=model,
33
+ >>> reco_arch='crnn_mobilenet_v3_small')
34
+
35
+ >>> # Get your predictions
36
+ >>> res = predictor(img)
37
+ ```
config.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ 3,
14
+ 32,
15
+ 128
16
+ ],
17
+ "vocab": "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ!\"#$%&'()*+,-./:;<=>?@[\\]^_`{|}~°àâéèêëîïôùûçÀÂÉÈËÎÏÔÙÛÇ£€¥¢฿",
18
+ "url": "https://github.com/felixdittrich92/OnnxTR/releases/download/v0.0.1/crnn_vgg16_bn-662979cc.onnx",
19
+ "url_8_bit": "https://github.com/felixdittrich92/OnnxTR/releases/download/v0.1.2/crnn_vgg16_bn_static_8_bit-bce050c7.onnx",
20
+ "arch": "crnn_vgg16_bn",
21
+ "task": "recognition"
22
+ }
model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:662979cc0d5ad13bdfb31f89daf8d02f521ddd5a10651a445e8e79918160447c
3
+ size 63210761