Felix92 commited on
Commit
e9a12e6
·
verified ·
1 Parent(s): b07c05c

Add text2knowledge/doctr-torch-tablecenternet model

Browse files
Files changed (3) hide show
  1. README.md +44 -0
  2. config.json +20 -0
  3. pytorch_model.bin +3 -0
README.md ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ tags:
4
+ - ocr
5
+ - pytorch
6
+ - doctr
7
+ - table_structure
8
+ ---
9
+
10
+
11
+ <p align="center">
12
+ <img src="https://doctr-static.mindee.com/models?id=v0.3.1/Logo_doctr.gif&src=0" width="60%">
13
+ </p>
14
+
15
+ **Optical Character Recognition made seamless & accessible to anyone, powered by PyTorch**
16
+
17
+ ## Task: table_structure
18
+
19
+ https://github.com/mindee/doctr
20
+
21
+ ### Example usage:
22
+
23
+ ```python
24
+ >>> from doctr.io import DocumentFile
25
+ >>> from doctr.models import ocr_predictor, from_hub
26
+
27
+ >>> img = DocumentFile.from_images(['<image_path>'])
28
+ >>> # Load your model from the hub
29
+ >>> model = from_hub('mindee/my-model')
30
+
31
+ >>> # Pass it to the predictor
32
+ >>> # If your model is a recognition model:
33
+ >>> predictor = ocr_predictor(det_arch='db_mobilenet_v3_large',
34
+ >>> reco_arch=model,
35
+ >>> pretrained=True)
36
+
37
+ >>> # If your model is a detection model:
38
+ >>> predictor = ocr_predictor(det_arch=model,
39
+ >>> reco_arch='crnn_mobilenet_v3_small',
40
+ >>> pretrained=True)
41
+
42
+ >>> # Get your predictions
43
+ >>> res = predictor(img)
44
+ ```
config.json ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "input_shape": [
3
+ 3,
4
+ 1024,
5
+ 1024
6
+ ],
7
+ "mean": [
8
+ 0.798,
9
+ 0.785,
10
+ 0.772
11
+ ],
12
+ "std": [
13
+ 0.264,
14
+ 0.2749,
15
+ 0.287
16
+ ],
17
+ "url": "https://github.com/mindee/doctr/releases/download/v1.0.1/tablecenternet-ea5b30a3.pt",
18
+ "arch": "tablecenternet",
19
+ "task": "table_structure"
20
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:041c3dba7cd8cc50c98b662b2fdb4a7f97d0f91a117283d7395c187747e30926
3
+ size 28606255