paddlex 3.0.0b2__py3-none-any.whl → 3.0.0rc0__py3-none-any.whl

paddlex/repo_manager/repo.py

@@ -33,6 +33,7 @@ from .utils import (
     switch_working_dir,
     to_dep_spec_pep508,
     env_marker_ast2expr,
+    install_external_deps,
 )
 
 __all__ = ["build_repo_instance", "build_repo_group_installer"]
@@ -114,15 +115,22 @@ class PPRepository(object):
         """install_deps"""
         return RepositoryGroupInstaller([self]).install_deps(*args, **kwargs)
 
-    def install_package(self, no_deps=False, clean=True):
+    def install_package(self, no_deps=False, clean=True, install_extra_only=False):
         """install_package"""
         editable = self.meta.get("editable", True)
         extra_editable = self.meta.get("extra_editable", None)
         if editable:
             logging.warning(f"{self.pkg_name} will be installed in editable mode.")
         with switch_working_dir(self.root_dir):
+            if install_extra_only:
+                src_requirements = os.path.join(self.root_dir, "requirements.txt")
+                paddlex_requirements = os.path.join(
+                    self.root_dir, "requirements_paddlex.txt"
+                )
+                shutil.copy(paddlex_requirements, src_requirements)
             try:
                 install_packages_using_pip(["."], editable=editable, no_deps=no_deps)
+                install_external_deps(self.name, self.root_dir)
             finally:
                 if clean:
                     # Clean build artifacts
@@ -209,10 +217,13 @@ class PPRepository(object):
         """get_pdx"""
         return importlib.import_module(self.pdx_mod_name)
 
-    def get_deps(self):
+    def get_deps(self, install_extra_only=False):
         """get_deps"""
         # Merge requirement files
-        req_list = [self.main_req_file]
+        if install_extra_only:
+            req_list = []
+        else:
+            req_list = [self.main_req_file]
         req_list.extend(self.meta.get("extra_req_files", []))
         deps = []
         for req in req_list:
@@ -268,7 +279,10 @@ class RepositoryGroupInstaller(object):
         # failure of one repo package aborts the entire installation process.
         for ins_flag, repo in zip(ins_flags, repos):
             if ins_flag:
-                repo.install_package(no_deps=True)
+                if repo.name in ["PaddleVideo"]:
+                    repo.install_package(no_deps=True, install_extra_only=True)
+                else:
+                    repo.install_package(no_deps=True)
 
     def uninstall(self):
         """uninstall"""
@@ -284,7 +298,10 @@ class RepositoryGroupInstaller(object):
         deps_list = []
         repos = self._sort_repos(self.repos, check_missing=True)
         for repo in repos:
-            deps = repo.get_deps()
+            if repo.name in ["PaddleVideo"]:
+                deps = repo.get_deps(install_extra_only=True)
+            else:
+                deps = repo.get_deps()
             deps = self._normalize_deps(deps, headline=f"# {repo.name} dependencies")
             deps_list.append(deps)
         # Add an extra new line to separate dependencies of different repos.
@@ -355,6 +372,8 @@ class RepositoryGroupInstaller(object):
             if n in repo_pkgs:
                 # Skip repo packages
                 continue
+            elif check_installation_using_pip(n):
+                continue
             else:
                 line_n = [n]
                 fe = f"[{','.join(e)}]" if e else ""

paddlex/repo_manager/requirements.txt

@@ -1,18 +1,21 @@
 tomark~=0.1.4
 psutil~=5.9.2
 wget
-grpcio==1.51.3
+grpcio==1.51.3; python_version<"3.12"
+grpcio==1.59.0; python_version>="3.12"
 pydantic>=2
 editdistance
 openpyxl
 premailer
 python-docx
 ######## For Chatocrv3 #######
-qianfan==0.0.3
-langchain==0.1.5
-langchain-community==0.0.17
-erniebot == 0.5.0
-erniebot-agent == 0.5.0
+langchain==0.2.17
+langchain-openai==0.1.25
+langchain-community==0.2.17
+langchain-text-splitters==0.2.4
+transformers==4.40.0
+openai==1.63.2
 unstructured
 networkx
-numpy==1.24.4
+numpy==1.24.4; python_version<"3.12"
+numpy==1.26.4; python_version>="3.12"

paddlex/repo_manager/utils.py

@@ -18,8 +18,11 @@ import json
 import platform
 import subprocess
 import contextlib
-
 from parsley import makeGrammar
+import lazy_paddle as paddle
+
+from ..utils.env import get_device_type
+from ..utils import logging
 
 
 PLATFORM = platform.system()
@@ -33,6 +36,35 @@ def _check_output(*args, **kwargs):
     return subprocess.check_output(*args, **kwargs)
 
 
+def _compare_version(version1, version2):
+    import re
+
+    def parse_version(version_str):
+        version_pattern = re.compile(
+            r"^(?P<major>\d+)\.(?P<minor>\d+)\.(?P<patch>\d+)(?:-(?P<pre_release>.*))?(?:\+(?P<build_metadata>.+))?$"
+        )
+        match = version_pattern.match(version_str)
+        if not match:
+            raise ValueError(f"Unexpected version string: {version_str}")
+        return (
+            int(match.group("major")),
+            int(match.group("minor")),
+            int(match.group("patch")),
+            match.group("pre_release"),
+        )
+
+    v1_infos = parse_version(version1)
+    v2_infos = parse_version(version2)
+    for v1_info, v2_info in zip(v1_infos, v2_infos):
+        if v1_info is None and v2_info is None:
+            continue
+        if v1_info is None or (v2_info is not None and v1_info < v2_info):
+            return -1
+        if v2_info is None or (v1_info is not None and v1_info > v2_info):
+            return 1
+    return 0
+
+
 def check_installation_using_pip(pkg):
     """check_installation_using_pip"""
     out = _check_output(["pip", "list", "--format", "json"])
@@ -69,6 +101,35 @@ def install_packages_using_pip(
     return _check_call(args)
 
 
+def install_external_deps(repo_name, repo_root):
+    """install paddle repository custom dependencies"""
+
+    def get_gcc_version():
+        return subprocess.check_output(["gcc", "--version"]).decode("utf-8").split()[2]
+
+    if repo_name == "PaddleDetection":
+        if os.path.exists(os.path.join(repo_root, "ppdet", "ext_op")):
+            """Install custom op for rotated object detection"""
+            if (
+                PLATFORM == "Linux"
+                and _compare_version(get_gcc_version(), "8.2.0") >= 0
+                and "gpu" in get_device_type()
+                and (
+                    paddle.is_compiled_with_cuda()
+                    and not paddle.is_compiled_with_rocm()
+                )
+            ):
+                with switch_working_dir(os.path.join(repo_root, "ppdet", "ext_op")):
+                    args = [sys.executable, "setup.py", "install"]
+                    _check_call(args)
+            else:
+                logging.warning(
+                    "The custom operators in PaddleDetection for Rotated Object Detection is only supported when using CUDA, GCC>=8.2.0 and Paddle>=2.0.1, "
+                    "your environment does not meet these requirements, so we will skip the installation of custom operators under PaddleDetection/ppdet/ext_ops, "
+                    "which means you can not train the Rotated Object Detection models."
+                )
+
+
 def install_deps_using_pip():
     """install requirements"""
     current_file_path = os.path.dirname(os.path.abspath(__file__))

paddlex/serving_requirements.txt

@@ -0,0 +1,9 @@
+aiohttp>=3.9
+bce-python-sdk>=0.9
+fastapi>=0.110
+filetype>=1.2
+pydantic>=2
+starlette>=0.36
+typing_extensions>=4.11
+uvicorn>=0.16
+yarl>=1.9

paddlex/utils/config.py

@@ -18,6 +18,7 @@ import argparse
 import yaml
 from . import logging
 from .errors import raise_key_not_found_error
+from .file_interface import custom_open
 
 __all__ = ["get_config"]
 
@@ -61,7 +62,7 @@ def create_attr_dict(yaml_config):
 
 def parse_config(cfg_file):
     """Load a config file into AttrDict"""
-    with open(cfg_file, "r") as fopen:
+    with custom_open(cfg_file, "r") as fopen:
         yaml_config = AttrDict(yaml.load(fopen, Loader=yaml.SafeLoader))
     create_attr_dict(yaml_config)
     return yaml_config
@@ -70,7 +71,7 @@ def parse_config(cfg_file):
 def print_dict(d, delimiter=0):
     """
     Recursively visualize a dict and
-    indenting acrrording by the relationship of keys.
+    indenting according by the relationship of keys.
     """
     placeholder = "-" * 60
     for k, v in sorted(d.items()):
@@ -124,7 +125,7 @@ def override(dl, ks, v):
             return s
 
     assert isinstance(dl, (list, dict)), "{} should be a list or a dict"
-    assert len(ks) > 0, "lenght of keys should larger than 0"
+    assert len(ks) > 0, "length of keys should larger than 0"
     if isinstance(dl, list):
         k = parse_str(ks[0])
         if len(ks) == 1:

paddlex/utils/custom_device_whitelist.py

@@ -0,0 +1,457 @@
+# copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
+#
+# 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.
+
+DCU_WHITELIST = [
+    "ResNet18",
+    "ResNet34",
+    "ResNet50",
+    "ResNet101",
+    "ResNet152",
+    "PP-LCNet_x1_0",
+    "PP-HGNetV2-B0_ML",
+    "PP-HGNetV2-B4_ML",
+    "PP-HGNetV2-B6_ML",
+    "CLIP_vit_base_patch16_224_ML",
+    "PP-ShiTuV2_rec_CLIP_vit_base",
+    "PP-YOLOE_plus-L",
+    "PP-YOLOE_plus-M",
+    "PP-YOLOE_plus-S",
+    "RT-DETR-R18",
+    "PicoDet-L",
+    "PicoDet-M",
+    "PicoDet-S",
+    "PicoDet-XS",
+    "FCOS-ResNet50",
+    "YOLOX-N",
+    "FasterRCNN-ResNet34-FPN",
+    "YOLOv3-DarkNet53",
+    "Cascade-FasterRCNN-ResNet50-FPN",
+    "PP-YOLOE_plus_SOD-S",
+    "PP-YOLOE_plus_SOD-L",
+    "PP-YOLOE_plus_SOD-largesize-L",
+    "STFPM",
+    "Deeplabv3_Plus-R50",
+    "Deeplabv3_Plus-R101",
+    "PP-LiteSeg-T",
+    "PP-OCRv4_server_rec",
+    "PP-OCRv4_server_det",
+    "PP-OCRv4_mobile_det",
+    "DLinear",
+    "RLinear",
+    "NLinear",
+    "PicoDet_LCNet_x2_5_face",
+]
+
+MLU_WHITELIST = [
+    "ResNet18",
+    "ResNet34",
+    "ResNet50",
+    "ResNet101",
+    "ResNet152",
+    "ResNet18_vd",
+    "ResNet34_vd",
+    "ResNet50_vd",
+    "ResNet101_vd",
+    "ResNet152_vd",
+    "ResNet200_vd",
+    "PP-LCNet_x0_25",
+    "PP-LCNet_x0_35",
+    "PP-LCNet_x0_5",
+    "PP-LCNet_x0_75",
+    "PP-LCNet_x1_0",
+    "PP-LCNet_x1_5",
+    "PP-LCNet_x2_5",
+    "PP-LCNet_x2_0",
+    "MobileNetV3_large_x0_35",
+    "MobileNetV3_large_x0_5",
+    "MobileNetV3_large_x0_75",
+    "MobileNetV3_large_x1_0",
+    "MobileNetV3_large_x1_25",
+    "MobileNetV3_small_x0_35",
+    "MobileNetV3_small_x0_5",
+    "MobileNetV3_small_x0_75",
+    "MobileNetV3_small_x1_0",
+    "MobileNetV3_small_x1_25",
+    "PP-HGNet_small",
+    "PP-HGNet_tiny",
+    "PP-HGNet_base",
+    "PP-ShiTuV2_rec_CLIP_vit_base",
+    "PP-ShiTuV2_rec_CLIP_vit_large",
+    "PP-YOLOE_plus-X",
+    "PP-YOLOE_plus-L",
+    "PP-YOLOE_plus-M",
+    "PP-YOLOE_plus-S",
+    "PicoDet-L",
+    "PicoDet-M",
+    "PicoDet-S",
+    "PicoDet-XS",
+    "STFPM",
+    "PP-LiteSeg-T",
+    "PP-OCRv4_server_rec",
+    "PP-OCRv4_mobile_rec",
+    "PP-OCRv4_server_det",
+    "PP-OCRv4_mobile_det",
+    "PicoDet_layout_1x",
+    "DLinear",
+    "RLinear",
+    "NLinear",
+    "PicoDet_LCNet_x2_5_face",
+]
+
+NPU_WHITELIST = [
+    "ResNet18",
+    "ResNet34",
+    "ResNet50",
+    "ResNet101",
+    "ResNet152",
+    "ResNet18_vd",
+    "ResNet34_vd",
+    "ResNet50_vd",
+    "ResNet101_vd",
+    "ResNet152_vd",
+    "ResNet200_vd",
+    "PP-LCNet_x0_25",
+    "PP-LCNet_x0_35",
+    "PP-LCNet_x0_5",
+    "PP-LCNet_x0_75",
+    "PP-LCNet_x1_0",
+    "PP-LCNet_x1_5",
+    "PP-LCNet_x2_5",
+    "PP-LCNet_x2_0",
+    "PP-LCNetV2_small",
+    "PP-LCNetV2_base",
+    "PP-LCNetV2_large",
+    "MobileNetV3_large_x0_35",
+    "MobileNetV3_large_x0_5",
+    "MobileNetV3_large_x0_75",
+    "MobileNetV3_large_x1_0",
+    "MobileNetV3_large_x1_25",
+    "MobileNetV3_small_x0_35",
+    "MobileNetV3_small_x0_5",
+    "MobileNetV3_small_x0_75",
+    "MobileNetV3_small_x1_0",
+    "MobileNetV3_small_x1_25",
+    "ConvNeXt_tiny",
+    "ConvNeXt_small",
+    "ConvNeXt_base_224",
+    "ConvNeXt_base_384",
+    "ConvNeXt_large_224",
+    "ConvNeXt_large_384",
+    "MobileNetV1_x0_25",
+    "MobileNetV1_x0_5",
+    "MobileNetV1_x0_75",
+    "MobileNetV1_x1_0",
+    "MobileNetV2_x0_25",
+    "MobileNetV2_x0_5",
+    "MobileNetV2_x1_0",
+    "MobileNetV2_x1_5",
+    "MobileNetV2_x2_0",
+    "SwinTransformer_tiny_patch4_window7_224",
+    "SwinTransformer_small_patch4_window7_224",
+    "SwinTransformer_base_patch4_window7_224",
+    "SwinTransformer_base_patch4_window12_384",
+    "SwinTransformer_large_patch4_window7_224",
+    "SwinTransformer_large_patch4_window12_384",
+    "PP-HGNet_small",
+    "PP-HGNet_tiny",
+    "PP-HGNet_base",
+    "PP-HGNetV2-B0",
+    "PP-HGNetV2-B1",
+    "PP-HGNetV2-B2",
+    "PP-HGNetV2-B3",
+    "PP-HGNetV2-B4",
+    "PP-HGNetV2-B5",
+    "PP-HGNetV2-B6",
+    "CLIP_vit_base_patch16_224",
+    "CLIP_vit_large_patch14_224",
+    "MobileNetV4_conv_small",
+    "MobileNetV4_conv_medium",
+    "MobileNetV4_hybrid_medium",
+    "MobileNetV4_conv_large",
+    "MobileNetV4_hybrid_large",
+    "StarNet-S1 ",
+    "StarNet-S2",
+    "StarNet-S3",
+    "StarNet-S4",
+    "FasterNet-T0",
+    "FasterNet-T1",
+    "FasterNet-T2",
+    "FasterNet-S",
+    "FasterNet-M",
+    "FasterNet-L",
+    "PP-LCNet_x1_0_ML",
+    "ResNet50_ML",
+    "PP-HGNetV2-B0_ML",
+    "PP-HGNetV2-B4_ML",
+    "PP-HGNetV2-B6_ML",
+    "CLIP_vit_base_patch16_224_ML",
+    "PPLCNet_x1_0_pedestrian_attribute",
+    "PP-LCNet_x1_0_vehicle_attribute",
+    "PP-ShiTuV2_rec",
+    "PP-ShiTuV2_rec_CLIP_vit_base",
+    "PP-ShiTuV2_rec_CLIP_vit_large",
+    "PP-YOLOE_plus-X",
+    "PP-YOLOE_plus-L",
+    "PP-YOLOE_plus-M",
+    "PP-YOLOE_plus-S",
+    "RT-DETR-L",
+    "RT-DETR-H",
+    "RT-DETR-X",
+    "RT-DETR-R18",
+    "RT-DETR-R50",
+    "PicoDet-L",
+    "PicoDet-M",
+    "PicoDet-S",
+    "PicoDet-XS",
+    "CenterNet-DLA-34",
+    "CenterNet-ResNet50",
+    "FCOS-ResNet50",
+    "DETR-R50",
+    "YOLOX-N",
+    "YOLOX-T",
+    "YOLOX-S",
+    "YOLOX-M",
+    "FasterRCNN-ResNet34-FPN",
+    "FasterRCNN-ResNet50",
+    "FasterRCNN-ResNet50-FPN",
+    "FasterRCNN-ResNet50-vd-FPN",
+    "FasterRCNN-ResNet50-vd-SSLDv2-FPN",
+    "FasterRCNN-ResNet101",
+    "FasterRCNN-ResNet101-FPN",
+    "FasterRCNN-ResNeXt101-vd-FPN",
+    "FasterRCNN-Swin-Tiny-FPN",
+    "YOLOv3-DarkNet53",
+    "YOLOv3-ResNet50_vd_DCN",
+    "YOLOv3-MobileNetV3",
+    "Cascade-FasterRCNN-ResNet50-FPN",
+    "Cascade-FasterRCNN-ResNet50-vd-SSLDv2-FPN",
+    "PP-ShiTuV2_det",
+    "PP-YOLOE-S_human",
+    "PP-YOLOE-L_human",
+    "PP-YOLOE-L_vehicle",
+    "PP-YOLOE-S_vehicle",
+    "PP-YOLOE_plus_SOD-S",
+    "PP-YOLOE_plus_SOD-L",
+    "PP-YOLOE_plus_SOD-largesize-L",
+    "STFPM",
+    "Deeplabv3-R50",
+    "Deeplabv3-R101",
+    "Deeplabv3_Plus-R50",
+    "Deeplabv3_Plus-R101",
+    "PP-LiteSeg-T",
+    "OCRNet_HRNet-W18",
+    "OCRNet_HRNet-W48",
+    "SeaFormer-tiny",
+    "SeaFormer-small",
+    "SeaFormer-base",
+    "SeaFormer-large",
+    "SegFormer-B0",
+    "SegFormer-B1",
+    "SegFormer-B2",
+    "SegFormer-B3",
+    "SegFormer-B4",
+    "SegFormer-B5",
+    "PP-YOLOE_seg-S",
+    "SOLOv2",
+    "MaskRCNN-ResNet50",
+    "MaskRCNN-ResNet50-FPN",
+    "MaskRCNN-ResNet50-vd-FPN",
+    "MaskRCNN-ResNet101-FPN",
+    "MaskRCNN-ResNet101-vd-FPN",
+    "MaskRCNN-ResNeXt101-vd-FPN",
+    "Cascade-MaskRCNN-ResNet50-vd-SSLDv2-FPN",
+    "Cascade-MaskRCNN-ResNet50-FPN",
+    "Mask-RT-DETR-H",
+    "Mask-RT-DETR-L",
+    "Mask-RT-DETR-X",
+    "Mask-RT-DETR-M",
+    "Mask-RT-DETR-S",
+    "PP-OCRv4_server_rec",
+    "PP-OCRv4_mobile_rec",
+    "PP-OCRv4_server_det",
+    "PP-OCRv4_mobile_det",
+    "ch_RepSVTR_rec",
+    "ch_SVTRv2_rec",
+    "PicoDet_layout_1x",
+    "PicoDet-L_layout_3cls",
+    "RT-DETR-H_layout_3cls",
+    "RT-DETR-H_layout_17cls",
+    "SLANet",
+    "SLANet_plus",
+    "DLinear",
+    "DLinear_ad",
+    "RLinear",
+    "NLinear",
+    "PatchTST",
+    "PatchTST_ad",
+    "TimesNet",
+    "TimesNet_cls",
+    "TimesNet_ad",
+    "TiDE",
+    "Nonstationary",
+    "Nonstationary_ad",
+    "AutoEncoder_ad",
+    "PicoDet_LCNet_x2_5_face",
+    "UVDoc",
+    "PP-OCRv4_mobile_seal_det",
+    "PP-OCRv4_server_seal_det",
+    "PP-LCNet_x1_0_doc_ori",
+]
+
+XPU_WHITELIST = [
+    "ResNet18",
+    "ResNet34",
+    "ResNet50",
+    "ResNet101",
+    "ResNet152",
+    "ResNet18_vd",
+    "ResNet34_vd",
+    "ResNet50_vd",
+    "ResNet101_vd",
+    "ResNet152_vd",
+    "ResNet200_vd",
+    "PP-LCNet_x0_25",
+    "PP-LCNet_x0_35",
+    "PP-LCNet_x0_5",
+    "PP-LCNet_x0_75",
+    "PP-LCNet_x1_0",
+    "PP-LCNet_x1_5",
+    "PP-LCNet_x2_5",
+    "PP-LCNet_x2_0",
+    "MobileNetV3_large_x0_35",
+    "MobileNetV3_large_x0_5",
+    "MobileNetV3_large_x0_75",
+    "MobileNetV3_large_x1_0",
+    "MobileNetV3_large_x1_25",
+    "MobileNetV3_small_x0_35",
+    "MobileNetV3_small_x0_5",
+    "MobileNetV3_small_x0_75",
+    "MobileNetV3_small_x1_0",
+    "MobileNetV3_small_x1_25",
+    "PP-HGNet_small",
+    "PP-HGNet_tiny",
+    "PP-HGNet_base",
+    "PP-YOLOE_plus-X",
+    "PP-YOLOE_plus-L",
+    "PP-YOLOE_plus-M",
+    "PP-YOLOE_plus-S",
+    "PicoDet-L",
+    "PicoDet-M",
+    "PicoDet-S",
+    "PicoDet-XS",
+    "STFPM",
+    "PP-LiteSeg-T",
+    "PP-OCRv4_server_rec",
+    "PP-OCRv4_mobile_rec",
+    "PP-OCRv4_server_det",
+    "PP-OCRv4_mobile_det",
+    "PicoDet_layout_1x",
+    "DLinear",
+    "RLinear",
+    "NLinear",
+    "PicoDet_LCNet_x2_5_face",
+]
+
+GCU_WHITELIST = [
+    "ConvNeXt_base_224",
+    "ConvNeXt_base_384",
+    "ConvNeXt_large_224",
+    "ConvNeXt_large_384",
+    "ConvNeXt_small",
+    "ConvNeXt_tiny",
+    "FasterNet-L",
+    "FasterNet-M",
+    "FasterNet-S",
+    "FasterNet-T0",
+    "FasterNet-T1",
+    "FasterNet-T2",
+    "MobileNetV1_x0_25",
+    "MobileNetV1_x0_5",
+    "MobileNetV1_x0_75",
+    "MobileNetV1_x1_0",
+    "MobileNetV2_x0_25",
+    "MobileNetV2_x0_5",
+    "MobileNetV2_x1_0",
+    "MobileNetV2_x1_5",
+    "MobileNetV2_x2_0",
+    "MobileNetV3_large_x0_35",
+    "MobileNetV3_large_x0_5",
+    "MobileNetV3_large_x0_75",
+    "MobileNetV3_large_x1_0",
+    "MobileNetV3_large_x1_25",
+    "MobileNetV3_small_x0_35",
+    "MobileNetV3_small_x0_5",
+    "MobileNetV3_small_x0_75",
+    "MobileNetV3_small_x1_0",
+    "MobileNetV3_small_x1_25",
+    "MobileNetV4_conv_large",
+    "MobileNetV4_conv_medium",
+    "MobileNetV4_conv_small",
+    "PP-HGNet_base",
+    "PP-HGNet_small",
+    "PP-HGNet_tiny",
+    "PP-HGNetV2-B0",
+    "PP-HGNetV2-B1",
+    "PP-HGNetV2-B2",
+    "PP-HGNetV2-B3",
+    "PP-HGNetV2-B4",
+    "PP-HGNetV2-B5",
+    "PP-HGNetV2-B6",
+    "PP-LCNet_x0_25",
+    "PP-LCNet_x0_35",
+    "PP-LCNet_x0_5",
+    "PP-LCNet_x0_75",
+    "PP-LCNet_x1_0",
+    "PP-LCNet_x1_5",
+    "PP-LCNet_x2_0",
+    "PP-LCNet_x2_5",
+    "PP-LCNetV2_base",
+    "PP-LCNetV2_large",
+    "PP-LCNetV2_small",
+    "ResNet18_vd",
+    "ResNet18",
+    "ResNet34_vd",
+    "ResNet34",
+    "ResNet50_vd",
+    "ResNet50",
+    "ResNet101_vd",
+    "ResNet101",
+    "ResNet152_vd",
+    "ResNet152",
+    "ResNet200_vd",
+    "StarNet-S1",
+    "StarNet-S2",
+    "StarNet-S3",
+    "StarNet-S4",
+    "FCOS-ResNet50",
+    "PicoDet-L",
+    "PicoDet-M",
+    "PicoDet-S",
+    "PicoDet-XS",
+    "PP-YOLOE_plus-L",
+    "PP-YOLOE_plus-M",
+    "PP-YOLOE_plus-S",
+    "PP-YOLOE_plus-X",
+    "RT-DETR-H",
+    "RT-DETR-L",
+    "RT-DETR-R18",
+    "RT-DETR-R50",
+    "RT-DETR-X",
+    "PP-YOLOE-L_human",
+    "PP-YOLOE-S_human",
+    "PP-OCRv4_mobile_det",
+    "PP-OCRv4_server_det",
+    "PP-OCRv4_mobile_rec",
+    "PP-OCRv4_server_rec",
+]