zhangyubo0722 9 months ago
parent
commit
96cf29b804

+ 2 - 2
paddlex/inference/pipelines/instance_segmentation/pipeline.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Union, Any, Dict, Optional, Union, List
+from typing import Any, Dict, Optional, Union, List
 import numpy as np
 from ...utils.pp_option import PaddlePredictorOption
 from ..base import BasePipeline
@@ -61,7 +61,7 @@ class InstanceSegmentationPipeline(BasePipeline):
 
         Args:
             input (str | list[str] | np.ndarray | list[np.ndarray]): The input image(s) or path(s) to the images.
-            threshold (float | None): The threshold value to filter out low-confidence predictions. Default is None.
+            threshold (Union[float, None]): The threshold value to filter out low-confidence predictions. Default is None.
             **kwargs: Additional keyword arguments that can be passed to the function.
 
         Returns:

+ 22 - 22
paddlex/inference/pipelines/layout_parsing/pipeline_v2.py

@@ -263,12 +263,12 @@ class LayoutParsingPipelineV2(BasePipeline):
 
     def get_model_settings(
         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:
         """
         Get the model settings based on the provided parameters or default values.
@@ -314,24 +314,24 @@ class LayoutParsingPipelineV2(BasePipeline):
     def predict(
         self,
         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_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_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_nms: Optional[bool] = None,
         layout_unclip_ratio: Optional[Union[float, Tuple[float, float]]] = None,