|
@@ -263,12 +263,12 @@ class LayoutParsingPipelineV2(BasePipeline):
|
|
|
|
|
|
|
|
def get_model_settings(
|
|
def get_model_settings(
|
|
|
self,
|
|
self,
|
|
|
- use_doc_orientation_classify: bool | None,
|
|
|
|
|
- use_doc_unwarping: bool | None,
|
|
|
|
|
- use_general_ocr: bool | None,
|
|
|
|
|
- use_seal_recognition: bool | None,
|
|
|
|
|
- use_table_recognition: bool | None,
|
|
|
|
|
- use_formula_recognition: bool | None,
|
|
|
|
|
|
|
+ use_doc_orientation_classify: Union[bool, None],
|
|
|
|
|
+ use_doc_unwarping: Union[bool, None],
|
|
|
|
|
+ use_general_ocr: Union[bool, None],
|
|
|
|
|
+ use_seal_recognition: Union[bool, None],
|
|
|
|
|
+ use_table_recognition: Union[bool, None],
|
|
|
|
|
+ use_formula_recognition: Union[bool, None],
|
|
|
) -> dict:
|
|
) -> dict:
|
|
|
"""
|
|
"""
|
|
|
Get the model settings based on the provided parameters or default values.
|
|
Get the model settings based on the provided parameters or default values.
|
|
@@ -314,24 +314,24 @@ class LayoutParsingPipelineV2(BasePipeline):
|
|
|
def predict(
|
|
def predict(
|
|
|
self,
|
|
self,
|
|
|
input: Union[str, list[str], np.ndarray, list[np.ndarray]],
|
|
input: Union[str, list[str], np.ndarray, list[np.ndarray]],
|
|
|
- use_doc_orientation_classify: bool | None = None,
|
|
|
|
|
- use_doc_unwarping: bool | None = None,
|
|
|
|
|
- use_general_ocr: bool | None = None,
|
|
|
|
|
- use_seal_recognition: bool | None = None,
|
|
|
|
|
- use_table_recognition: bool | None = None,
|
|
|
|
|
- use_formula_recognition: bool | None = None,
|
|
|
|
|
- text_det_limit_side_len: int | None = None,
|
|
|
|
|
|
|
+ use_doc_orientation_classify: Union[bool, None] = None,
|
|
|
|
|
+ use_doc_unwarping: Union[bool, None] = None,
|
|
|
|
|
+ use_general_ocr: Union[bool, None] = None,
|
|
|
|
|
+ use_seal_recognition: Union[bool, None] = None,
|
|
|
|
|
+ use_table_recognition: Union[bool, None] = None,
|
|
|
|
|
+ use_formula_recognition: Union[bool, None] = None,
|
|
|
|
|
+ text_det_limit_side_len: Union[int, None] = None,
|
|
|
text_det_limit_type: Union[str, None] = None,
|
|
text_det_limit_type: Union[str, None] = None,
|
|
|
- text_det_thresh: float | None = None,
|
|
|
|
|
- text_det_box_thresh: float | None = None,
|
|
|
|
|
- text_det_unclip_ratio: float | None = None,
|
|
|
|
|
- text_rec_score_thresh: float | None = None,
|
|
|
|
|
- seal_det_limit_side_len: int | None = None,
|
|
|
|
|
|
|
+ text_det_thresh: Union[float, None] = None,
|
|
|
|
|
+ text_det_box_thresh: Union[float, None] = None,
|
|
|
|
|
+ text_det_unclip_ratio: Union[float, None] = None,
|
|
|
|
|
+ text_rec_score_thresh: Union[float, None] = None,
|
|
|
|
|
+ seal_det_limit_side_len: Union[int, None] = None,
|
|
|
seal_det_limit_type: Union[str, None] = None,
|
|
seal_det_limit_type: Union[str, None] = None,
|
|
|
- seal_det_thresh: float | None = None,
|
|
|
|
|
- seal_det_box_thresh: float | None = None,
|
|
|
|
|
- seal_det_unclip_ratio: float | None = None,
|
|
|
|
|
- seal_rec_score_thresh: float | None = None,
|
|
|
|
|
|
|
+ seal_det_thresh: Union[float, None] = None,
|
|
|
|
|
+ seal_det_box_thresh: Union[float, None] = None,
|
|
|
|
|
+ seal_det_unclip_ratio: Union[float, None] = None,
|
|
|
|
|
+ seal_rec_score_thresh: Union[float, None] = None,
|
|
|
layout_threshold: Optional[Union[float, dict]] = None,
|
|
layout_threshold: Optional[Union[float, dict]] = None,
|
|
|
layout_nms: Optional[bool] = None,
|
|
layout_nms: Optional[bool] = None,
|
|
|
layout_unclip_ratio: Optional[Union[float, Tuple[float, float]]] = None,
|
|
layout_unclip_ratio: Optional[Union[float, Tuple[float, float]]] = None,
|