ReadingPipeline-notebooks / pipeline_config.json
Stanislav Kalinin
feat: Update school notebooks models
ce41b2a
raw
history blame
1.29 kB
{ "main_process": {
"ImageToBGR": {
"input_format": "BGR"
},
"SegmPrediction": {
"model_path": "segm/segm_model.ckpt",
"config_path": "segm/segm_config.json",
"num_threads": 8,
"device": "cuda",
"runtime": "Pytorch"
},
"RestoreImageAngle": {
"restoring_class_names": ["text_line"]
},
"ClassContourPosptrocess": {},
"OCRPrediction": {
"model_path": "ocr/ocr_model.ckpt",
"config_path": "ocr/ocr_config.json",
"lm_path": "ocr/kenlm_corpus.arpa",
"num_threads": 8,
"classes_to_ocr": ["shrinked_text"],
"device": "cuda",
"batch_size": 128,
"runtime": "Pytorch"
},
"LineFinder": {
"line_classes": ["text_line"],
"text_classes": ["shrinked_text"]
},
"PrepareJSON": {}
},
"classes": {
"shrinked_text": {
"contour_posptrocess": {
"BboxFromContour": {},
"UpscaleBbox": {"upscale_bbox": [1.4, 2.3]},
"CropByBbox": {}
}
},
"text_line": {
"contour_posptrocess": {}
}
}
}