-
Notifications
You must be signed in to change notification settings - Fork 7
/
Copy pathpipeline_config.json
44 lines (44 loc) · 1.13 KB
/
pipeline_config.json
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
{ "main_process": {
"ImageToBGR": {
"input_format": "BGR"
},
"SegmPrediction": {
"model_path": "",
"config_path": "",
"num_threads": 8,
"device": "cuda",
"runtime": "Pytorch"
},
"RestoreImageAngle": {
"restoring_class_names": ["text_line"]
},
"ClassContourPosptrocess": {},
"OCRPrediction": {
"model_path": "",
"config_path": "",
"lm_path": "",
"num_threads": 8,
"classes_to_ocr": ["text"],
"device": "cuda",
"batch_size": 64,
"runtime": "Pytorch"
},
"LineFinder": {
"line_classes": ["text_line"],
"text_classes": ["text"]
},
"PrepareJSON": {}
},
"classes": {
"text": {
"contour_posptrocess": {
"BboxFromContour": {},
"UpscaleBbox": {"upscale_bbox": [1.4, 2.3]},
"CropByBbox": {}
}
},
"text_line": {
"contour_posptrocess": {}
}
}
}