Bingsu commited on
Commit
d1d37ff
1 Parent(s): 8ab5436

Add doctr_db_resnet50-detection-210116 model

Browse files
Files changed (3) hide show
  1. README.md +39 -0
  2. config.json +20 -0
  3. pytorch_model.bin +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: detection
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,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://doctr-static.mindee.com/models?id=v0.3.1/db_resnet50-ac60cadc.pt&src=0",
18
+ "arch": "db_resnet50",
19
+ "task": "detection"
20
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:177ef868dc284d7078c49e5d7cd1d4f1b3b8a034e8c17e71ba08f5a3275a7aef
3
+ size 102007541