{ "main_process": { | |
"ImageToBGR": { | |
"input_format": "BGR" | |
}, | |
"SegmPrediction": { | |
"model_path": "segm/model-100-0.1701.ckpt", | |
"config_path": "segm/segm_config.json", | |
"device": "cuda" | |
}, | |
"RestoreImageAngle": { | |
"restoring_class_names": ["shrinked_text"] | |
}, | |
"ClassContourPosptrocess": {}, | |
"OCRPrediction": { | |
"model_path": "ocr/model-95-0.0293.ckpt", | |
"config_path": "ocr/ocr_config.json", | |
"lm_path": "", | |
"classes_to_ocr": ["shrinked_text"], | |
"device": "cuda" | |
}, | |
"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": {} | |
} | |
} | |
} | |
} | |