paddlex 3.0.0rc1__py3-none-any.whl → 3.0.2__py3-none-any.whl

paddlex/repo_manager/repo.py

@@ -28,7 +28,7 @@ from ..utils.install import (
     install_packages_from_requirements_file,
     uninstall_packages,
 )
-from .meta import REPO_DOWNLOAD_BASE, get_repo_meta
+from .meta import REPO_DIST_NAMES, REPO_DOWNLOAD_BASE, get_repo_meta
 from .utils import (
     fetch_repo_using_git,
     install_external_deps,
@@ -133,8 +133,10 @@ class PPRepository(object):
                 try:
                     pip_install_opts = ["--no-deps"]
                     if editable:
-                        pip_install_opts.append("-e")
-                    install_packages(["."], pip_install_opts=pip_install_opts)
+                        reqs = ["-e ."]
+                    else:
+                        reqs = ["."]
+                    install_packages(reqs, pip_install_opts=pip_install_opts)
                     install_external_deps(self.name, self.root_dir)
                 finally:
                     if clean:
@@ -142,14 +144,16 @@ class PPRepository(object):
                         tmp_build_dir = "build"
                         if osp.exists(tmp_build_dir):
                             shutil.rmtree(tmp_build_dir)
-        for e in self.meta.get("extra", []):
+        for e in self.meta.get("extra_pkgs", []):
             if isinstance(e, tuple):
                 with switch_working_dir(osp.join(self.root_dir, e[0])):
+                    pip_install_opts = ["--no-deps"]
+                    if e[3]:
+                        reqs = ["-e ."]
+                    else:
+                        reqs = ["."]
                     try:
-                        pip_install_opts = ["--no-deps"]
-                        if e[3]:
-                            pip_install_opts.append("-e")
-                        install_packages(["."], pip_install_opts=pip_install_opts)
+                        install_packages(reqs, pip_install_opts=pip_install_opts)
                     finally:
                         if clean:
                             tmp_build_dir = "build"
@@ -159,9 +163,9 @@ class PPRepository(object):
     def uninstall_packages(self):
         """uninstall_packages"""
         pkgs = []
-        if self.install_pkg:
+        if self.meta["install_pkg"]:
             pkgs.append(self.dist_name)
-        for e in self.meta.get("extra", []):
+        for e in self.meta.get("extra_pkgs", []):
             if isinstance(e, tuple):
                 pkgs.append(e[1])
         uninstall_packages(pkgs)
@@ -206,7 +210,7 @@ class PPRepository(object):
         """get_deps"""
         # Merge requirement files
         req_list = [self.main_req_file]
-        for e in self.meta.get("extra", []):
+        for e in self.meta.get("extra_pkgs", []):
             if isinstance(e, tuple):
                 e = e[2] or osp.join(e[0], "requirements.txt")
             req_list.append(e)
@@ -327,7 +331,7 @@ class RepositoryGroupInstaller(object):
 
     def _sort_repos(self, repos, check_missing=False):
         # We sort the repos to ensure that the dependencies precede the
-        # dependant in the list.
+        # dependent in the list.
         name_meta_pairs = []
         for repo in repos:
             name_meta_pairs.append((repo.name, repo.meta))
@@ -358,9 +362,6 @@ class RepositoryGroupInstaller(object):
         return sorted_repos
 
     def _normalize_deps(self, deps, headline=None):
-        repo_pkgs = set(
-            repo.dist_name for repo in self.repos if repo.dist_name is not None
-        )
         lines = []
         if headline is not None:
             lines.append(headline)
@@ -376,7 +377,7 @@ class RepositoryGroupInstaller(object):
             # If `line` is not an empty line or a comment, it must be a requirement specifier.
             # Other forms may cause a parse error.
             req = Requirement(line_s)
-            if req.name in repo_pkgs:
+            if req.name in REPO_DIST_NAMES:
                 # Skip repo packages
                 continue
             elif req.name.replace("_", "-") in (

paddlex/utils/custom_device_list.py

@@ -135,8 +135,6 @@ NPU_BLACKLIST = [
     "whisper_base",
     "whisper_tiny",
     "UniMERNet",
-    "PP-FormulaNet-L",
-    "PP-FormulaNet-S",
 ]
 
 XPU_WHITELIST = [
@@ -191,6 +189,32 @@ XPU_WHITELIST = [
     "RLinear",
     "NLinear",
     "PicoDet_LCNet_x2_5_face",
+    "PP-LCNet_x1_0_doc_ori",
+    "UVDoc",
+    "PP-DocLayout-L",
+    "PP-DocLayout-M",
+    "PP-DocLayout-S",
+    "PP-DocLayout_plus-L",
+    "PP-DocBlockLayout",
+    "SLANeXt_wired",
+    "SLANeXt_wireless",
+    "PP-LCNet_x1_0_table_cls",
+    "RT-DETR-L_wired_table_cell_det",
+    "RT-DETR-L_wireless_table_cell_det",
+    "PP-OCRv4_server_seal_det",
+    "PP-OCRv4_mobile_seal_det",
+    "PP-OCRv5_server_rec",
+    "PP-OCRv5_mobile_rec",
+    "PPDocBee-2B",
+    "PPDocBee-7B",
+    "PP-FormulaNet-S",
+    "PP-FormulaNet-L",
+    "PP-FormulaNet-M",
+    "PP-FormulaNet_plus-M",
+    "PP-FormulaNet_plus-L",
+    "PP-OCRv5_server_det",
+    "PP-OCRv5_mobile_det",
+    "PP-Chart2Table",
 ]
 
 GCU_WHITELIST = [

paddlex/utils/deps.py

@@ -73,7 +73,7 @@ def _get_dep_specs():
 DEP_SPECS = _get_dep_specs()
 
 
-def _get_dep_version(dep):
+def get_dep_version(dep):
     try:
         return importlib.metadata.version(dep)
     except importlib.metadata.PackageNotFoundError:
@@ -101,7 +101,7 @@ def is_dep_available(dep, /, check_version=None):
             check_version = True
         else:
             check_version = False
-    version = _get_dep_version(dep)
+    version = get_dep_version(dep)
     if version is None:
         return False
     if check_version:
@@ -246,4 +246,4 @@ def require_paddle2onnx_plugin():
 
 
 def get_paddle2onnx_spec():
-    return "paddle2onnx >= 2.0.1"
+    return "paddle2onnx == 2.0.2rc3"

paddlex/utils/device.py

@@ -15,8 +15,6 @@
 import os
 from contextlib import ContextDecorator
 
-import GPUtil
-
 from . import logging
 from .custom_device_list import (
     DCU_WHITELIST,
@@ -41,18 +39,12 @@ def constr_device(device_type, device_ids):
 
 
 def get_default_device():
-    avail_gpus = GPUtil.getAvailable()
-    if not avail_gpus:
-        # maybe edge devices like Jetson
-        if os.path.exists("/etc/nv_tegra_release"):
-            avail_gpus = [0]
-            logging.info(
-                "Detected that the current device is a Jetson edge device. The default behavior will be to use GPU: 0"
-            )
-    if not avail_gpus:
-        return "cpu"
+    import paddle
+
+    if paddle.device.is_compiled_with_cuda() and paddle.device.cuda.device_count() > 0:
+        return constr_device("gpu", [0])
     else:
-        return constr_device("gpu", [avail_gpus[0]])
+        return "cpu"
 
 
 def parse_device(device):

paddlex/utils/env.py

@@ -41,6 +41,8 @@ def get_paddle_cuda_version():
     import paddle.version
 
     cuda_version = paddle.version.cuda()
+    if cuda_version == "False":
+        return None
     return tuple(map(int, cuda_version.split(".")))
 
 
@@ -48,6 +50,8 @@ def get_paddle_cudnn_version():
     import paddle.version
 
     cudnn_version = paddle.version.cudnn()
+    if cudnn_version == "False":
+        return None
     return tuple(map(int, cudnn_version.split(".")))
 
 

paddlex/utils/flags.py

@@ -24,7 +24,6 @@ __all__ = [
     "INFER_BENCHMARK_ITERS",
     "INFER_BENCHMARK_WARMUP",
     "INFER_BENCHMARK_OUTPUT_DIR",
-    "INFER_BENCHMARK_USE_NEW_INFER_API",
     "FLAGS_json_format_model",
     "USE_PIR_TRT",
     "DISABLE_DEV_MODEL_WL",
@@ -52,6 +51,17 @@ FLAGS_json_format_model = get_flag_from_env_var("FLAGS_json_format_model", True)
 USE_PIR_TRT = get_flag_from_env_var("PADDLE_PDX_USE_PIR_TRT", True)
 DISABLE_DEV_MODEL_WL = get_flag_from_env_var("PADDLE_PDX_DISABLE_DEV_MODEL_WL", False)
 DISABLE_CINN_MODEL_WL = get_flag_from_env_var("PADDLE_PDX_DISABLE_CINN_MODEL_WL", False)
+DISABLE_TRT_MODEL_BL = get_flag_from_env_var("PADDLE_PDX_DISABLE_TRT_MODEL_BL", False)
+DISABLE_MKLDNN_MODEL_BL = get_flag_from_env_var(
+    "PADDLE_PDX_DISABLE_MKLDNN_MODEL_BL", False
+)
+LOCAL_FONT_FILE_PATH = get_flag_from_env_var("PADDLE_PDX_LOCAL_FONT_FILE_PATH", None)
+ENABLE_MKLDNN_BYDEFAULT = get_flag_from_env_var(
+    "PADDLE_PDX_ENABLE_MKLDNN_BYDEFAULT", True
+)
+
+MODEL_SOURCE = os.environ.get("PADDLE_PDX_MODEL_SOURCE", "huggingface")
+
 
 # Inference Benchmark
 INFER_BENCHMARK = get_flag_from_env_var("PADDLE_PDX_INFER_BENCHMARK", False)
@@ -67,6 +77,3 @@ INFER_BENCHMARK_ITERS = get_flag_from_env_var(
 INFER_BENCHMARK_USE_CACHE_FOR_READ = get_flag_from_env_var(
     "PADDLE_PDX_INFER_BENCHMARK_USE_CACHE_FOR_READ", False
 )
-INFER_BENCHMARK_USE_NEW_INFER_API = get_flag_from_env_var(
-    "PADDLE_PDX_INFER_BENCHMARK_USE_NEW_INFER_API", False
-)

paddlex/utils/fonts/__init__.py

@@ -18,7 +18,9 @@ import PIL
 from PIL import ImageFont
 
 from .. import logging
+from ..cache import CACHE_DIR
 from ..download import download
+from ..flags import LOCAL_FONT_FILE_PATH
 
 
 def get_font_file_path(file_name: str) -> str:
@@ -28,8 +30,8 @@ def get_font_file_path(file_name: str) -> str:
     Returns:
     str: The path to the font file.
     """
-    font_path = (Path(__file__).parent / file_name).resolve().as_posix()
-    if not Path(font_path).exists():
+    font_path = (Path(CACHE_DIR) / "fonts" / file_name).resolve().as_posix()
+    if not Path(font_path).is_file():
         download(
             url=f"https://paddle-model-ecology.bj.bcebos.com/paddlex/PaddleX3.0/fonts/{file_name}",
             save_path=font_path,
@@ -64,5 +66,33 @@ def create_font(txt: str, sz: tuple, font_path: str) -> ImageFont:
     return font
 
 
-PINGFANG_FONT_FILE_PATH = get_font_file_path("PingFang-SC-Regular.ttf")
-SIMFANG_FONT_FILE_PATH = get_font_file_path("simfang.ttf")
+def create_font_vertical(
+    txt: str, sz: tuple, font_path: str, scale=1.2
+) -> ImageFont.FreeTypeFont:
+    n = len(txt) if len(txt) > 0 else 1
+    base_font_size = int(sz[1] / n * 0.8 * scale)
+    base_font_size = max(base_font_size, 10)
+    font = ImageFont.truetype(font_path, base_font_size, encoding="utf-8")
+
+    if int(PIL.__version__.split(".")[0]) < 10:
+        max_char_width = max([font.getsize(c)[0] for c in txt])
+    else:
+        max_char_width = max([font.getlength(c) for c in txt])
+
+    if max_char_width > sz[0]:
+        new_size = int(base_font_size * sz[0] / max_char_width)
+        new_size = max(new_size, 10)
+        font = ImageFont.truetype(font_path, new_size, encoding="utf-8")
+
+    return font
+
+
+if Path(str(LOCAL_FONT_FILE_PATH)).is_file():
+    logging.warning(
+        f"Using the local font file(`{LOCAL_FONT_FILE_PATH}`) specified by `LOCAL_FONT_FILE_PATH`!"
+    )
+    PINGFANG_FONT_FILE_PATH = LOCAL_FONT_FILE_PATH
+    SIMFANG_FONT_FILE_PATH = LOCAL_FONT_FILE_PATH
+else:
+    PINGFANG_FONT_FILE_PATH = get_font_file_path("PingFang-SC-Regular.ttf")
+    SIMFANG_FONT_FILE_PATH = get_font_file_path("simfang.ttf")

paddlex/utils/misc.py

@@ -70,7 +70,7 @@ class CachedProperty(object):
     https://github.com/pydanny/cached-property/blob/master/cached_property.py .
 
     Note that this implementation does NOT work in multi-thread or coroutine
-    senarios.
+    scenarios.
     """
 
     def __init__(self, func):

paddlex/utils/subclass_register.py

@@ -46,7 +46,7 @@ class AutoRegisterMetaClass(type):
         if bases:
             for base in bases:
                 base_cls = mcs.__find_base_class(base)
-                if base_cls:
+                if base_cls and hasattr(cls, mcs.__model_type_attr_name):
                     mcs.__register_to_base_class(base_cls, cls)
 
     @classmethod
@@ -64,7 +64,7 @@ class AutoRegisterMetaClass(type):
 
     @classmethod
     def __register_to_base_class(mcs, base, cls):
-        cls_entity_name = getattr(cls, mcs.__model_type_attr_name, cls.__name__)
+        cls_entity_name = getattr(cls, mcs.__model_type_attr_name)
         if isinstance(cls_entity_name, str):
             cls_entity_name = [cls_entity_name]