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

paddlex/modules/multilabel_classification/dataset_checker/dataset_src/convert_dataset.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,17 +14,21 @@
 
 
 import os
-import json
-from tqdm import tqdm
-from pycocotools.coco import COCO
+
+from .....utils.deps import function_requires_deps, is_dep_available
 from .....utils.errors import ConvertFailedError
 from .....utils.logging import info, warning
 
+if is_dep_available("pycocotools"):
+    from pycocotools.coco import COCO
+if is_dep_available("tqdm"):
+    from tqdm import tqdm
+
 
 def check_src_dataset(root_dir, dataset_type):
     """check src dataset format validity"""
     if dataset_type in ("COCO"):
-        anno_suffix = ".json"
+        pass
     else:
         raise ConvertFailedError(
             message=f"数据格式转换失败！不支持{dataset_type}格式数据集。当前仅支持 COCO 格式。"
@@ -61,6 +65,7 @@ def convert_coco_dataset(root_dir):
         coco2multilabels(src_img_dir, src_anno_path, root_dir)
 
 
+@function_requires_deps("pycocotools", "tqdm")
 def coco2multilabels(src_img_dir, src_anno_path, root_dir):
     image_dir = os.path.join(root_dir, "images")
     label_type = (
@@ -74,7 +79,6 @@ def coco2multilabels(src_img_dir, src_anno_path, root_dir):
     num_classes = len(list(cat_id_map.keys()))
 
     with open(anno_save_path, "w") as fp:
-        lines = []
         for img_id in tqdm(sorted(coco.getImgIds())):
             img_info = coco.loadImgs([img_id])[0]
             img_filename = img_info["file_name"]
@@ -108,7 +112,6 @@ def coco2multilabels(src_img_dir, src_anno_path, root_dir):
     if label_type == "train":
         label_txt_save_path = os.path.join(root_dir, "label.txt")
         with open(label_txt_save_path, "w") as fp:
-            label_name_list = []
             for cat in coco.cats.values():
                 id = cat["id"]
                 name = cat["name"]

paddlex/modules/multilabel_classification/dataset_checker/dataset_src/split_dataset.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/modules/multilabel_classification/dataset_checker/dataset_src/utils/__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.

paddlex/modules/multilabel_classification/dataset_checker/dataset_src/utils/visualizer.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,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import os
 import numpy as np
-import json
-from pathlib import Path
-import PIL
 from PIL import Image, ImageDraw, ImageFont
 
 from ......utils.fonts import PINGFANG_FONT_FILE_PATH

paddlex/modules/multilabel_classification/evaluator.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/modules/multilabel_classification/exportor.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/modules/multilabel_classification/model_list.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/modules/multilabel_classification/trainer.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,13 +12,11 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import json
 import shutil
 from pathlib import Path
 
 from ..base import BaseTrainer
 from .model_list import MODELS
-from ...utils.config import AttrDict
 
 
 class MLClsTrainer(BaseTrainer):
@@ -82,4 +80,6 @@ class MLClsTrainer(BaseTrainer):
         ):
             train_args["resume_path"] = self.train_config.resume_path
         train_args["dy2st"] = self.train_config.get("dy2st", False)
+        # amp support 'O1', 'O2', 'OFF'
+        train_args["amp"] = self.train_config.get("amp", "OFF")
         return train_args

paddlex/modules/multilingual_speech_recognition/__init__.py

@@ -0,0 +1,18 @@
+# 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.
+
+from .dataset_checker import WhisperDatasetChecker
+from .evaluator import WhisperEvaluator
+from .exportor import WhisperExportor
+from .trainer import WhisperTrainer

paddlex/modules/multilingual_speech_recognition/dataset_checker.py

@@ -0,0 +1,27 @@
+# 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.
+
+from ...utils.errors import UnsupportedAPIError
+from ..base import BaseDatasetChecker
+from .model_list import MODELS
+
+
+class WhisperDatasetChecker(BaseDatasetChecker):
+    """Dataset Checker for Whisper Model"""
+
+    entities = MODELS
+
+    def __init__(self, config):
+        # not support for now
+        raise UnsupportedAPIError("whisper model not support data check for now.")

paddlex/modules/multilingual_speech_recognition/evaluator.py

@@ -0,0 +1,27 @@
+# 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.
+
+from ...utils.errors import UnsupportedAPIError
+from ..base import BaseEvaluator
+from .model_list import MODELS
+
+
+class WhisperEvaluator(BaseEvaluator):
+    """Instance Whisper Model Evaluator"""
+
+    entities = MODELS
+
+    def __init__(self, config):
+        # not support for now
+        raise UnsupportedAPIError("whisper model not support evaluate for now.")

paddlex/modules/multilingual_speech_recognition/exportor.py

@@ -0,0 +1,27 @@
+# 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.
+
+from ...utils.errors import UnsupportedAPIError
+from ..base import BaseExportor
+from .model_list import MODELS
+
+
+class WhisperExportor(BaseExportor):
+    """Instance Whisper Model Exportor"""
+
+    entities = MODELS
+
+    def __init__(self, config):
+        # not support for now
+        raise UnsupportedAPIError("whisper model not support export for now.")

paddlex/modules/multilingual_speech_recognition/model_list.py

@@ -0,0 +1,22 @@
+# 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.
+
+
+MODELS = [
+    "whisper_large",
+    "whisper_medium",
+    "whisper_base",
+    "whisper_small",
+    "whisper_tiny",
+]

paddlex/modules/multilingual_speech_recognition/trainer.py

@@ -0,0 +1,42 @@
+# 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.
+
+from ...utils.errors import UnsupportedAPIError
+from ..base import BaseTrainer
+from .model_list import MODELS
+
+
+class WhisperTrainer(BaseTrainer):
+    """Instance Whisper Model Trainer"""
+
+    entities = MODELS
+
+    def __init__(self, config):
+        # not support for now
+        raise UnsupportedAPIError("whisper model not support train for now.")
+
+    def update_config(self):
+        """update training config"""
+
+    def get_train_kwargs(self) -> dict:
+        """get key-value arguments of model training function
+
+        Returns:
+            dict: the arguments of training function.
+        """
+        train_args = {
+            "device": self.get_device(),
+            "amp": self.train_config.get("amp", "OFF"),  # amp support 'O1', 'O2', 'OFF'
+        }
+        return train_args

paddlex/modules/object_detection/__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.
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from .trainer import DetTrainer
 from .dataset_checker import COCODatasetChecker
 from .evaluator import DetEvaluator
 from .exportor import DetExportor
+from .trainer import DetTrainer

paddlex/modules/object_detection/dataset_checker/__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.
@@ -12,20 +12,11 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-
-import os
-import os.path as osp
 from pathlib import Path
-from collections import defaultdict, Counter
-
-from PIL import Image
-import json
-from pycocotools.coco import COCO
 
 from ...base import BaseDatasetChecker
-from .dataset_src import check, convert, split_dataset, deep_analyse
-
 from ..model_list import MODELS
+from .dataset_src import check, convert, deep_analyse, split_dataset
 
 
 class COCODatasetChecker(BaseDatasetChecker):

paddlex/modules/object_detection/dataset_checker/dataset_src/__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.
@@ -13,7 +13,7 @@
 # limitations under the License.
 
 
+from .analyse_dataset import deep_analyse
 from .check_dataset import check
 from .convert_dataset import convert
 from .split_dataset import split_dataset
-from .analyse_dataset import deep_analyse

paddlex/modules/object_detection/dataset_checker/dataset_src/analyse_dataset.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,20 +14,22 @@
 
 
 import os
-import json
 import platform
-from pathlib import Path
 from collections import defaultdict
-from PIL import Image
+
 import numpy as np
-import matplotlib.pyplot as plt
-from matplotlib import font_manager
-from matplotlib.backends.backend_agg import FigureCanvasAgg
-from pycocotools.coco import COCO
 
+from .....utils.deps import function_requires_deps, is_dep_available
 from .....utils.fonts import PINGFANG_FONT_FILE_PATH
 
+if is_dep_available("matplotlib"):
+    import matplotlib.pyplot as plt
+    from matplotlib import font_manager
+if is_dep_available("pycocotools"):
+    from pycocotools.coco import COCO
+
 
+@function_requires_deps("pycocotools", "matplotlib")
 def deep_analyse(dataset_dir, output):
     """class analysis for dataset"""
     tags = ["train", "val"]

paddlex/modules/object_detection/dataset_checker/dataset_src/check_dataset.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,18 +13,23 @@
 # limitations under the License.
 
 
+import json
 import os
 import os.path as osp
-from collections import defaultdict, Counter
+from collections import Counter, defaultdict
 from pathlib import Path
+
 from PIL import Image, ImageOps
-import json
-from pycocotools.coco import COCO
 
+from .....utils.deps import function_requires_deps, is_dep_available
 from .....utils.errors import DatasetFileNotFoundError
 from .utils.visualizer import draw_bbox
 
+if is_dep_available("pycocotools"):
+    from pycocotools.coco import COCO
+
 
+@function_requires_deps("pycocotools")
 def check(dataset_dir, output, sample_num=10):
     """check dataset"""
     dataset_dir = osp.abspath(dataset_dir)
@@ -33,7 +38,7 @@ def check(dataset_dir, output, sample_num=10):
 
     sample_cnts = dict()
     sample_paths = defaultdict(list)
-    im_sizes = defaultdict(Counter)
+    defaultdict(Counter)
     tags = ["instance_train", "instance_val"]
     for _, tag in enumerate(tags):
         file_list = osp.join(dataset_dir, f"annotations/{tag}.json")

paddlex/modules/object_detection/dataset_checker/dataset_src/convert_dataset.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,17 +13,20 @@
 # limitations under the License.
 
 
-import os
-import shutil
 import json
+import os
 import random
+import shutil
 import xml.etree.ElementTree as ET
-from tqdm import tqdm
 
-from .....utils.file_interface import custom_open, write_json_file
+from .....utils.deps import function_requires_deps, is_dep_available
 from .....utils.errors import ConvertFailedError
+from .....utils.file_interface import custom_open, write_json_file
 from .....utils.logging import info, warning
 
+if is_dep_available("tqdm"):
+    from tqdm import tqdm
+
 
 class Indexer(object):
     """Indexer"""
@@ -69,9 +72,9 @@ class Extension(object):
 def check_src_dataset(root_dir, dataset_type):
     """check src dataset format validity"""
     if dataset_type in ("VOC", "VOCWithUnlabeled"):
-        anno_suffix = ".xml"
+        pass
     elif dataset_type in ("LabelMe", "LabelMeWithUnlabeled"):
-        anno_suffix = ".json"
+        pass
     else:
         raise ConvertFailedError(
             message=f"数据格式转换失败！不支持{dataset_type}格式数据集。当前仅支持 VOC、LabelMe 和 VOCWithUnlabeled、LabelMeWithUnlabeled 格式。"
@@ -168,6 +171,7 @@ def convert_labelme_dataset(root_dir, anno_map):
         )
 
 
+@function_requires_deps("tqdm")
 def labelme2coco(label_indexer, img_indexer, root_dir, anno_path, save_path):
     """convert json files generated by LabelMe to coco format and save to files"""
     with custom_open(anno_path, "r") as f:
@@ -305,7 +309,7 @@ def voc_get_image_info(annotation_root, img_indexer):
     """
     filename = annotation_root.findtext("filename")
     assert filename is not None, filename
-    img_name = os.path.basename(filename)
+    os.path.basename(filename)
     im_id = img_indexer.get_id(filename)
 
     size = annotation_root.find("size")
@@ -357,6 +361,7 @@ def voc_get_coco_annotation(obj, label_indexer):
     return anno
 
 
+@function_requires_deps("tqdm")
 def voc_xmls_to_cocojson(
     root_dir, annotation_paths, label_indexer, img_indexer, output, output_file
 ):

paddlex/modules/object_detection/dataset_checker/dataset_src/split_dataset.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,15 +13,18 @@
 # limitations under the License.
 
 
+import json
 import os
-import shutil
 import random
-import json
-from tqdm import tqdm
+import shutil
 
+from .....utils.deps import function_requires_deps, is_dep_available
 from .....utils.file_interface import custom_open, write_json_file
 from .....utils.logging import info
 
+if is_dep_available("tqdm"):
+    from tqdm import tqdm
+
 
 def split_dataset(root_dir, train_rate, val_rate):
     """split dataset"""
@@ -70,6 +73,7 @@ def split_dataset(root_dir, train_rate, val_rate):
     return root_dir
 
 
+@function_requires_deps("tqdm")
 def json2list(json_path, base_image_num):
     """load json as list"""
     assert os.path.exists(json_path), json_path

paddlex/modules/object_detection/dataset_checker/dataset_src/utils/__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.

paddlex/modules/object_detection/dataset_checker/dataset_src/utils/visualizer.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,16 +12,16 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import os
 import numpy as np
-import json
-from pathlib import Path
 import PIL
-from PIL import Image, ImageDraw, ImageFont
-from pycocotools.coco import COCO
+from PIL import ImageDraw, ImageFont
 
-from ......utils.fonts import PINGFANG_FONT_FILE_PATH
 from ......utils import logging
+from ......utils.deps import function_requires_deps, is_dep_available
+from ......utils.fonts import PINGFANG_FONT_FILE_PATH
+
+if is_dep_available("pycocotools"):
+    from pycocotools.coco import COCO
 
 
 def colormap(rgb=False):
@@ -114,7 +114,8 @@ def font_colormap(color_index):
         return dark.astype("int32")
 
 
-def draw_bbox(image, coco_info: COCO, img_id):
+@function_requires_deps("pycocotools")
+def draw_bbox(image, coco_info: "COCO", img_id):
     """
     Draw bbox on image
     """

paddlex/modules/object_detection/evaluator.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.
@@ -22,11 +22,27 @@ class DetEvaluator(BaseEvaluator):
 
     entities = MODELS
 
+    def _update_dataset(self):
+        """update dataset settings"""
+        metric = self.pdx_config.metric if "metric" in self.pdx_config else "COCO"
+        data_fields = (
+            self.pdx_config.EvalDataset["data_fields"]
+            if "data_fields" in self.pdx_config.EvalDataset
+            else None
+        )
+
+        self.pdx_config.update_dataset(
+            self.global_config.dataset_dir,
+            "COCODetDataset",
+            data_fields=data_fields,
+            metric=metric,
+        )
+
     def update_config(self):
         """update evalution config"""
         if self.eval_config.log_interval:
             self.pdx_config.update_log_interval(self.eval_config.log_interval)
-        self.pdx_config.update_dataset(self.global_config.dataset_dir, "COCODetDataset")
+        self._update_dataset()
         self.pdx_config.update_weights(self.eval_config.weight_path)
 
     def get_eval_kwargs(self) -> dict:

paddlex/modules/object_detection/exportor.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/modules/object_detection/model_list.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.
@@ -71,4 +71,14 @@ MODELS = [
     "BlazeFace",
     "BlazeFace-FPN-SSH",
     "PP-YOLOE_plus-S_face",
+    "PP-YOLOE-R-L",
+    "Co-Deformable-DETR-R50",
+    "Co-Deformable-DETR-Swin-T",
+    "Co-DINO-R50",
+    "Co-DINO-Swin-L",
+    "RT-DETR-L_wired_table_cell_det",
+    "RT-DETR-L_wireless_table_cell_det",
+    "PP-DocLayout-L",
+    "PP-DocLayout-M",
+    "PP-DocLayout-S",
 ]