Stanislav Kalinin commited on
Commit
7b0a0eb
1 Parent(s): 71d126b

feat: Update pipeline_config.json

Browse files
Files changed (1) hide show
  1. 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"],