paddlex 3.0.0rc0__py3-none-any.whl → 3.0.1__py3-none-any.whl

paddlex/utils/fonts/__init__.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -16,8 +16,11 @@ from pathlib import Path
 
 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:
@@ -27,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,
@@ -63,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/func_register.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

paddlex/utils/install.py

@@ -0,0 +1,87 @@
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import os
+import subprocess
+import sys
+import tempfile
+
+from packaging.requirements import Requirement
+
+from . import logging
+
+
+def install_packages_from_requirements_file(
+    requirements_file_path, pip_install_opts=None
+):
+    from .deps import DEP_SPECS
+
+    # TODO: Precompute or cache the constraints
+    with tempfile.NamedTemporaryFile("w", suffix=".txt", delete=False) as f:
+        for reqs in DEP_SPECS.values():
+            for req in reqs:
+                req = Requirement(req)
+                if req.marker and not req.marker.evaluate():
+                    continue
+                if req.url:
+                    req = f"{req.name}@{req.url}"
+                else:
+                    req = f"{req.name}{req.specifier}"
+                f.write(req + "\n")
+        constraints_file_path = f.name
+
+    args = [
+        sys.executable,
+        "-m",
+        "pip",
+        "install",
+        "-c",
+        constraints_file_path,
+        *(pip_install_opts or []),
+        "-r",
+        requirements_file_path,
+    ]
+    logging.debug("Command: %s", args)
+
+    try:
+        return subprocess.check_call(args)
+    finally:
+        os.unlink(constraints_file_path)
+
+
+def install_packages(requirements, pip_install_opts=None):
+    with tempfile.NamedTemporaryFile("w", suffix=".txt", delete=False) as f:
+        for req in requirements:
+            f.write(req + "\n")
+        reqs_file_path = f.name
+    try:
+        return install_packages_from_requirements_file(
+            reqs_file_path, pip_install_opts=pip_install_opts
+        )
+    finally:
+        os.unlink(reqs_file_path)
+
+
+def uninstall_packages(packages, pip_uninstall_opts=None):
+    args = [
+        sys.executable,
+        "-m",
+        "pip",
+        "uninstall",
+        "-y",
+        *(pip_uninstall_opts or []),
+        *packages,
+    ]
+    logging.debug("Command: %s", args)
+    return subprocess.check_call(args)

paddlex/utils/interactive_get_pipeline.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -12,11 +12,11 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from pathlib import Path
 import shutil
+from pathlib import Path
 
-from ..utils import logging
 from ..inference.utils.get_pipeline_path import get_pipeline_path
+from ..utils import logging
 
 
 def interactive_get_pipeline(pipeline, save_path):

paddlex/utils/lazy_loader.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -14,9 +14,9 @@
 
 # Code copied from https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/util/lazy_loader.py
 import importlib
-import types
-import os
 import inspect
+import os
+import types
 
 from . import logging
 from .flags import FLAGS_json_format_model

paddlex/utils/logging.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -13,6 +13,7 @@
 # limitations under the License.
 
 
+import functools
 import inspect
 import logging
 import sys
@@ -67,6 +68,14 @@ def warning(msg, *args, **kwargs):
     _logger.warning(msg, *args, **kwargs)
 
 
+@functools.lru_cache(None)
+def warning_once(msg, *args, **kwargs):
+    """
+    This method is identical to `logger.warning()`, but will emit the warning with the same message only once
+    """
+    warning(msg, *args, **kwargs)
+
+
 def error(msg, *args, **kwargs):
     """error"""
     _logger.error(msg, *args, **kwargs)

paddlex/utils/misc.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -15,12 +15,14 @@
 
 import os
 import threading
-import numpy as np
 from abc import ABCMeta
+
+import numpy as np
+
 from .errors import (
+    DuplicateRegistrationError,
     raise_class_not_found_error,
     raise_no_entity_registered_error,
-    DuplicateRegistrationError,
 )
 from .logging import *
 
@@ -68,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):
@@ -197,5 +199,3 @@ class AutoRegisterMetaClass(type):
 
 class AutoRegisterABCMetaClass(ABCMeta, AutoRegisterMetaClass):
     """AutoRegisterABCMetaClass"""
-
-    pass

paddlex/utils/pipeline_arguments.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -13,9 +13,9 @@
 # limitations under the License.
 
 from ast import literal_eval
+from typing import Dict, List, Literal, Optional, Tuple, Union
+
 from pydantic import TypeAdapter, ValidationError
-from functools import wraps
-from typing import Dict, List, Tuple, Union, Literal, Optional
 
 
 def custom_type(cli_expected_type):
@@ -294,7 +294,9 @@ PIPELINE_ARGUMENTS = {
         },
         {
             "name": "--layout_unclip_ratio",
-            "type": custom_type(Optional[Union[float, Tuple[float, float], Dict[int, Tuple]]]),
+            "type": custom_type(
+                Optional[Union[float, Tuple[float, float], Dict[int, Tuple]]]
+            ),
             "help": "Determines unclip ratio for layout detection boxes",
         },
         {
@@ -381,7 +383,9 @@ PIPELINE_ARGUMENTS = {
         },
         {
             "name": "--layout_unclip_ratio",
-            "type": custom_type(Optional[Union[float, Tuple[float, float], Dict[int, Tuple]]]),
+            "type": custom_type(
+                Optional[Union[float, Tuple[float, float], Dict[int, Tuple]]]
+            ),
             "help": "Determines unclip ratio for layout detection boxes",
         },
         {
@@ -498,7 +502,9 @@ PIPELINE_ARGUMENTS = {
         },
         {
             "name": "--layout_unclip_ratio",
-            "type": custom_type(Optional[Union[float, Tuple[float, float], Dict[int, Tuple]]]),
+            "type": custom_type(
+                Optional[Union[float, Tuple[float, float], Dict[int, Tuple]]]
+            ),
             "help": "Determines unclip ratio for layout detection boxes",
         },
         {
@@ -613,7 +619,9 @@ PIPELINE_ARGUMENTS = {
         },
         {
             "name": "--layout_unclip_ratio",
-            "type": custom_type(Optional[Union[float, Tuple[float, float], Dict[int, Tuple]]]),
+            "type": custom_type(
+                Optional[Union[float, Tuple[float, float], Dict[int, Tuple]]]
+            ),
             "help": "Sets the layout unclip ratio for layout detection.",
         },
         {

paddlex/utils/result_saver.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -15,10 +15,9 @@
 
 import os
 import sys
-import json
 import traceback
-from .flags import DEBUG
-from .file_interface import custom_open, write_json_file
+
+from .file_interface import write_json_file
 
 
 def try_except_decorator(func):
@@ -29,7 +28,7 @@ def try_except_decorator(func):
             result = func(self, *args, **kwargs)
             if result:
                 save_result(True, self._mode, self._output, result_dict=result)
-        except Exception as e:
+        except Exception:
             exc_type, exc_value, exc_tb = sys.exc_info()
             save_result(
                 False,

paddlex/utils/subclass_register.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -16,9 +16,9 @@ from abc import ABCMeta
 
 from . import logging
 from .errors import (
+    DuplicateRegistrationError,
     raise_class_not_found_error,
     raise_no_entity_registered_error,
-    DuplicateRegistrationError,
 )
 
 
@@ -97,5 +97,3 @@ class AutoRegisterMetaClass(type):
 
 class AutoRegisterABCMetaClass(ABCMeta, AutoRegisterMetaClass):
     """AutoRegisterABCMetaClass"""
-
-    pass

paddlex/version.py

@@ -1,4 +1,4 @@
-# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -30,6 +30,7 @@ def get_pdx_version():
 def get_version_dict():
     """get_version_dict"""
     import paddle
+
     from . import repo_manager
 
     ver_dict = dict()