Stanislav Kalinin
commited on
Commit
·
7b0a0eb
1
Parent(s):
71d126b
feat: Update pipeline_config.json
Browse files- pipeline_config.json +5 -2
pipeline_config.json
CHANGED
@@ -5,7 +5,8 @@
|
|
5 |
"SegmPrediction": {
|
6 |
"model_path": "segm/segm_model.ckpt",
|
7 |
"config_path": "segm/segm_config.json",
|
8 |
-
"device": "cuda"
|
|
|
9 |
},
|
10 |
"RestoreImageAngle": {
|
11 |
"restoring_class_names": ["shrinked_text"]
|
@@ -16,7 +17,9 @@
|
|
16 |
"config_path": "ocr/ocr_config.json",
|
17 |
"lm_path": "",
|
18 |
"classes_to_ocr": ["shrinked_text"],
|
19 |
-
"device": "cuda"
|
|
|
|
|
20 |
},
|
21 |
"LineFinder": {
|
22 |
"line_classes": ["shrinked_text"],
|
|
|
5 |
"SegmPrediction": {
|
6 |
"model_path": "segm/segm_model.ckpt",
|
7 |
"config_path": "segm/segm_config.json",
|
8 |
+
"device": "cuda",
|
9 |
+
"onnx": false
|
10 |
},
|
11 |
"RestoreImageAngle": {
|
12 |
"restoring_class_names": ["shrinked_text"]
|
|
|
17 |
"config_path": "ocr/ocr_config.json",
|
18 |
"lm_path": "",
|
19 |
"classes_to_ocr": ["shrinked_text"],
|
20 |
+
"device": "cuda",
|
21 |
+
"batch_size": 64,
|
22 |
+
"onnx": false
|
23 |
},
|
24 |
"LineFinder": {
|
25 |
"line_classes": ["shrinked_text"],
|