ReadingPipeline-Peter / pipeline_config.json
sberbank-ai
Update pipeline_config.json
c0eea04
{ "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": ["shrinked_text"]
},
"ClassContourPosptrocess": {},
"OCRPrediction": {
"model_path": "ocr/ocr_model.ckpt",
"config_path": "ocr/ocr_config.json",
"lm_path": "",
"num_threads": 8,
"classes_to_ocr": ["shrinked_text"],
"device": "cuda",
"batch_size": 64,
"runtime": "Pytorch"
},
"LineFinder": {
"line_classes": ["shrinked_text"],
"text_classes": ["shrinked_text"],
"pages_clust_dist": 0.35
},
"PrepareJSON": {}
},
"classes": {
"shrinked_text": {
"contour_posptrocess": {
"UpscaleContour": {"upscale_contour": 1.45},
"BboxFromContour": {},
"CropByBbox": {},
"MakeMaskedCrop": {},
"RotateVerticalCrops": {}
}
}
}
}