layout_parsing.py 2.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374
  1. # copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
  2. #
  3. # Licensed under the Apache License, Version 2.0 (the "License");
  4. # you may not use this file except in compliance with the License.
  5. # You may obtain a copy of the License at
  6. #
  7. # http://www.apache.org/licenses/LICENSE-2.0
  8. #
  9. # Unless required by applicable law or agreed to in writing, software
  10. # distributed under the License is distributed on an "AS IS" BASIS,
  11. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. # See the License for the specific language governing permissions and
  13. # limitations under the License.
  14. from typing import Dict, Final, List, Optional, Union
  15. from pydantic import BaseModel, Field
  16. from typing_extensions import Annotated, Literal
  17. from ..infra.models import DataInfo, PrimaryOperations
  18. from .shared import ocr
  19. __all__ = [
  20. "INFER_ENDPOINT",
  21. "InferRequest",
  22. "LayoutParsingResult",
  23. "InferResult",
  24. "PRIMARY_OPERATIONS",
  25. ]
  26. INFER_ENDPOINT: Final[str] = "/layout-parsing"
  27. class InferRequest(ocr.BaseInferRequest):
  28. useDocOrientationClassify: Optional[bool] = None
  29. useDocUnwarping: Optional[bool] = None
  30. useGeneralOcr: Optional[bool] = None
  31. useSealRecognition: Optional[bool] = None
  32. useTableRecognition: Optional[bool] = None
  33. useFormulaRecognition: Optional[bool] = None
  34. textDetLimitSideLen: Optional[int] = None
  35. textDetLimitType: Optional[Literal["min", "max"]] = None
  36. textDetThresh: Optional[float] = None
  37. textDetBoxThresh: Optional[float] = None
  38. textDetUnclipRatio: Optional[float] = None
  39. textRecScoreThresh: Optional[float] = None
  40. sealDetLimitSideLen: Optional[int] = None
  41. sealDetLimitType: Optional[Literal["min", "max"]] = None
  42. sealDetThresh: Optional[float] = None
  43. sealDetBoxThresh: Optional[float] = None
  44. sealDetUnclipRatio: Optional[float] = None
  45. sealRecScoreThresh: Optional[float] = None
  46. layoutThreshold: Optional[float] = None
  47. layoutNms: Optional[bool] = None
  48. layoutUnclipRatio: Optional[
  49. Union[float, Annotated[List[float], Field(min_length=2, max_length=2)]]
  50. ] = None
  51. layoutMergeBboxesMode: Optional[Literal["union", "large", "small"]] = None
  52. class LayoutParsingResult(BaseModel):
  53. prunedResult: dict
  54. outputImages: Optional[Dict[str, str]] = None
  55. inputImage: Optional[str] = None
  56. class InferResult(BaseModel):
  57. layoutParsingResults: List[LayoutParsingResult]
  58. dataInfo: DataInfo
  59. PRIMARY_OPERATIONS: Final[PrimaryOperations] = {
  60. "infer": (INFER_ENDPOINT, InferRequest, InferResult),
  61. }