ReadingPipeline-notebooks / pipeline_config.json
Stanislav Kalinin
chore: Remove tetardi folder
24ec5ae
raw history blame
No virus
1.43 kB
{ "main_process": {
"ImageToBGR": {
"input_format": "BGR"
},
"SegmPrediction": {
"model_path": "segm/model-319-0.1578.ckpt",
"config_path": "segm/segm_config.json",
"device": "cuda"
},
"RestoreImageAngle": {
"restoring_class_names": ["text_line"]
},
"ClassContourPosptrocess": {},
"OCRPrediction": {
"model_path": "ocr/model-97-0.8179.ckpt",
"config_path": "ocr/ocr_config.json",
"lm_path": "ocr/corpus_o6.arpa",
"classes_to_ocr": ["shrinked_pupil_text", "shrinked_comments"],
"device": "cuda"
},
"LineFinder": {
"line_classes": ["text_line"],
"text_classes": ["shrinked_pupil_text", "shrinked_comments"]
},
"PrepareJSON": {}
},
"classes": {
"shrinked_pupil_text": {
"contour_posptrocess": {
"BboxFromContour": {},
"UpscaleBbox": {"upscale_bbox": [1.4, 2.3]},
"CropByBbox": {}
}
},
"shrinked_comments": {
"contour_posptrocess": {
"BboxFromContour": {},
"UpscaleBbox": {"upscale_bbox": [1.4, 2.3]},
"CropByBbox": {}
}
},
"text_line": {
"contour_posptrocess": {}
}
}
}