diff --git a/paddlex/utils/flags.py b/paddlex/utils/flags.py index 7d304a52b..760296828 100644 --- a/paddlex/utils/flags.py +++ b/paddlex/utils/flags.py @@ -58,6 +58,7 @@ def get_flag_from_env_var(name, default, format_func=str): "PADDLE_PDX_DISABLE_MKLDNN_MODEL_BL", False ) LOCAL_FONT_FILE_PATH = get_flag_from_env_var("PADDLE_PDX_LOCAL_FONT_FILE_PATH", None) +FONTS_REMOTE_REPOSITORY_URL = os.environ.get("PADDLE_PDX_FONTS_REMOTE_REPOSITORY_URL", "https://paddle-model-ecology.bj.bcebos.com/paddlex/PaddleX3.0/fonts") ENABLE_MKLDNN_BYDEFAULT = get_flag_from_env_var( "PADDLE_PDX_ENABLE_MKLDNN_BYDEFAULT", True ) diff --git a/paddlex/utils/fonts.py b/paddlex/utils/fonts.py index c268b3a52..0459176af 100644 --- a/paddlex/utils/fonts.py +++ b/paddlex/utils/fonts.py @@ -20,7 +20,7 @@ from . import logging from .cache import CACHE_DIR from .download import download -from .flags import LOCAL_FONT_FILE_PATH +from .flags import LOCAL_FONT_FILE_PATH, FONTS_REMOTE_REPOSITORY_URL def create_font(txt: str, sz: tuple, font_path: str) -> ImageFont: @@ -94,7 +94,7 @@ def _get_offical_font(self): font_path = (Path(CACHE_DIR) / "fonts" / self._font_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/{self._font_name}", + url=f"${FONTS_REMOTE_REPOSITORY_URL}/{self._font_name}", save_path=font_path, ) self._local_path = font_path