ReadingPipeline-Peter / pipeline_config.json
Stanislav Kalinin
feat: Add weights and configs for Peter models
54c5290
raw
history blame
1.18 kB
{ "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": {}
}
}
}
}