__init__.py 3.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103
  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 pathlib import Path
  15. from ..utils.official_models import official_models
  16. from .base import BasePredictor, BasicPredictor
  17. from .image_classification import ClasPredictor
  18. from .text_detection import TextDetPredictor
  19. from .text_recognition import TextRecPredictor
  20. from .table_recognition import TablePredictor
  21. from .object_detection import DetPredictor
  22. from .instance_segmentation import InstanceSegPredictor
  23. from .semantic_segmentation import SegPredictor
  24. from .general_recognition import ShiTuRecPredictor
  25. from .ts_fc import TSFcPredictor
  26. from .ts_ad import TSAdPredictor
  27. from .ts_cls import TSClsPredictor
  28. from .image_unwarping import WarpPredictor
  29. from .multilabel_classification import MLClasPredictor
  30. from .anomaly_detection import UadPredictor
  31. def _create_hp_predictor(
  32. model_name, model_dir, device, config, hpi_params, *args, **kwargs
  33. ):
  34. try:
  35. from paddlex_hpi.predictors import HPPredictor
  36. except ModuleNotFoundError as e:
  37. raise RuntimeError(
  38. "The PaddleX HPI plugin is not properly installed, and the high-performance model inference features are not available."
  39. )
  40. if hpi_params is None:
  41. raise ValueError("No HPI params given")
  42. if "serial_number" not in hpi_params:
  43. raise ValueError("The serial number is required but was not provided.")
  44. serial_number = hpi_params["serial_number"]
  45. update_license = hpi_params.get("update_license", False)
  46. return HPPredictor.get(model_name)(
  47. model_dir=model_dir,
  48. config=config,
  49. device=device,
  50. serial_number=serial_number,
  51. update_license=update_license,
  52. *args,
  53. **kwargs,
  54. )
  55. def create_predictor(
  56. model: str,
  57. device=None,
  58. pp_option=None,
  59. use_hpip=False,
  60. hpi_params=None,
  61. *args,
  62. **kwargs,
  63. ) -> BasePredictor:
  64. model_dir = check_model(model)
  65. config = BasePredictor.load_config(model_dir)
  66. model_name = config["Global"]["model_name"]
  67. if use_hpip:
  68. return _create_hp_predictor(
  69. model_name=model_name,
  70. model_dir=model_dir,
  71. config=config,
  72. hpi_params=hpi_params,
  73. device=device,
  74. *args,
  75. **kwargs,
  76. )
  77. else:
  78. return BasicPredictor.get(model_name)(
  79. model_dir=model_dir,
  80. config=config,
  81. device=device,
  82. pp_option=pp_option,
  83. *args,
  84. **kwargs,
  85. )
  86. def check_model(model):
  87. if Path(model).exists():
  88. return Path(model)
  89. elif model in official_models:
  90. return official_models[model]
  91. else:
  92. raise Exception(
  93. f"The model ({model}) is no exists! Please using directory of local model files or model name supported by PaddleX!"
  94. )