Alexleetw commited on
Commit
63993b5
1 Parent(s): 4ecef07

Add db_resnet50_20240430-025744 model

Browse files
Files changed (3) hide show
  1. README.md +70 -0
  2. config.json +20 -0
  3. pytorch_model.bin +3 -0
README.md ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ ```
40
+ ### Run Configuration
41
+
42
+ {
43
+ "train_path": "/workspace/donut_train/doctr/train/",
44
+ "val_path": "/workspace/donut_train/doctr/val/",
45
+ "arch": "db_resnet50",
46
+ "name": null,
47
+ "epochs": 5,
48
+ "batch_size": 10,
49
+ "device": 0,
50
+ "save_interval_epoch": false,
51
+ "input_size": 1024,
52
+ "lr": 0.001,
53
+ "weight_decay": 0,
54
+ "workers": 16,
55
+ "resume": null,
56
+ "test_only": false,
57
+ "freeze_backbone": false,
58
+ "show_samples": false,
59
+ "wb": false,
60
+ "push_to_hub": true,
61
+ "pretrained": false,
62
+ "rotation": false,
63
+ "eval_straight": false,
64
+ "sched": "poly",
65
+ "amp": false,
66
+ "find_lr": false,
67
+ "early_stop": false,
68
+ "early_stop_epochs": 5,
69
+ "early_stop_delta": 0.01
70
+ }
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.7.0/db_resnet50-79bd7d70.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:af284ffa332e0c8a557531cf055aa8d13e46f57a863f0ba433941eaaff412e93
3
+ size 102013490