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

paddlex/modules/instance_segmentation/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.
@@ -15,10 +15,9 @@
 import os
 from pathlib import Path
 
-from .dataset_src import check, convert, split_dataset, deep_analyse
 from ...base import BaseDatasetChecker
-
 from ..model_list import MODELS
+from .dataset_src import check, convert, deep_analyse, split_dataset
 
 
 class COCOInstSegDatasetChecker(BaseDatasetChecker):

paddlex/modules/instance_segmentation/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/instance_segmentation/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.
@@ -16,16 +16,20 @@
 import os
 import platform
 from collections import defaultdict
-from pathlib import Path
 
-import matplotlib.pyplot as plt
 import numpy as np
-from matplotlib import font_manager
-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/instance_segmentation/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.
@@ -15,19 +15,22 @@
 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
-from pycocotools.coco import COCO
 
-from .utils.visualizer import draw_bbox, draw_mask
+from .....utils.deps import function_requires_deps
 from .....utils.errors import DatasetFileNotFoundError
 from .....utils.logging import info
+from .utils.visualizer import draw_bbox, draw_mask
 
 
+@function_requires_deps("pycocotools")
 def check(dataset_dir, output, sample_num=10):
     """check dataset"""
+    from pycocotools.coco import COCO
+
     info(dataset_dir)
     dataset_dir = osp.abspath(dataset_dir)
     if not osp.exists(dataset_dir) or not osp.isdir(dataset_dir):
@@ -35,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/instance_segmentation/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,19 +13,17 @@
 # limitations under the License.
 
 
-import os
-import shutil
 import json
+import os
 import random
-import xml.etree.ElementTree as ET
+import shutil
 
-import cv2
 import numpy as np
 from PIL import Image, ImageDraw
-from tqdm import tqdm
 
-from .....utils.file_interface import custom_open, write_json_file
+from .....utils.deps import function_requires_deps
 from .....utils.errors import ConvertFailedError
+from .....utils.file_interface import custom_open, write_json_file
 from .....utils.logging import info, warning
 
 
@@ -73,7 +71,7 @@ class Extension(object):
 def check_src_dataset(root_dir, dataset_type):
     """check src dataset format validity"""
     if dataset_type == "LabelMe":
-        anno_suffix = ".json"
+        pass
     else:
         raise ConvertFailedError(
             message=f"数据格式转换失败！不支持{dataset_type}格式数据集。当前仅支持 LabelMe 格式。"
@@ -164,9 +162,11 @@ def convert_labelme_dataset(root_dir, anno_map):
         )
 
 
+@function_requires_deps("tqdm", "pycocotools")
 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"""
     import pycocotools.mask as mask_util
+    from tqdm import tqdm
 
     with custom_open(anno_path, "r") as f:
         json_list = f.readlines()

paddlex/modules/instance_segmentation/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,12 +13,12 @@
 # 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
 from .....utils.file_interface import custom_open, write_json_file
 from .....utils.logging import info
 
@@ -70,8 +70,11 @@ 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"""
+    from tqdm import tqdm
+
     assert os.path.exists(json_path), json_path
     with custom_open(json_path, "r") as f:
         data = json.load(f)

paddlex/modules/instance_segmentation/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/instance_segmentation/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 ......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
     """
@@ -191,7 +192,8 @@ def draw_bbox(image, coco_info: COCO, img_id):
     return image
 
 
-def draw_mask(image, coco_info: COCO, img_id):
+@function_requires_deps("pycocotools")
+def draw_mask(image, coco_info: "COCO", img_id):
     """
     Draw mask on image
     """

paddlex/modules/instance_segmentation/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.
@@ -23,7 +23,7 @@ class InstanceSegEvaluator(DetEvaluator):
     entities = MODELS
 
     def update_config(self):
-        """update evalution config"""
+        """update evaluation config"""
         if self.eval_config.log_interval:
             self.pdx_config.update_log_interval(self.eval_config.log_interval)
         self.pdx_config.update_dataset(

paddlex/modules/instance_segmentation/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/instance_segmentation/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/instance_segmentation/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.

paddlex/modules/keypoint_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 KeypointTrainer
 from .dataset_checker import KeypointDatasetChecker
 from .evaluator import KeypointEvaluator
 from .exportor import KeypointExportor
+from .trainer import KeypointTrainer

paddlex/modules/keypoint_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.
@@ -14,8 +14,8 @@
 
 
 from ...object_detection.dataset_checker import COCODatasetChecker
-from .dataset_src import check
 from ..model_list import MODELS
+from .dataset_src import check
 
 
 class KeypointDatasetChecker(COCODatasetChecker):

paddlex/modules/keypoint_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.

paddlex/modules/keypoint_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_keypoint
 
+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/keypoint_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/keypoint_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.
@@ -13,10 +13,15 @@
 # limitations under the License.
 import numpy as np
 from PIL import ImageDraw
-from pycocotools.coco import COCO
 
+from ......utils.deps import function_requires_deps, is_dep_available
 
-def draw_keypoint(image, coco_info: COCO, img_id):
+if is_dep_available("pycocotools"):
+    from pycocotools.coco import COCO
+
+
+@function_requires_deps("pycocotools")
+def draw_keypoint(image, coco_info: "COCO", img_id):
     """
     Draw keypoints on image for the COCO human keypoint dataset with 17 keypoints.
 

paddlex/modules/keypoint_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.
@@ -23,7 +23,7 @@ class KeypointEvaluator(DetEvaluator):
     entities = MODELS
 
     def update_config(self):
-        """update evalution config"""
+        """update evaluation config"""
         if self.eval_config.log_interval:
             self.pdx_config.update_log_interval(self.eval_config.log_interval)
         metric = self.pdx_config.metric

paddlex/modules/keypoint_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/keypoint_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.

paddlex/modules/keypoint_detection/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.
@@ -13,8 +13,8 @@
 # limitations under the License.
 
 
-from .model_list import MODELS
 from ..object_detection import DetTrainer
+from .model_list import MODELS
 
 
 class KeypointTrainer(DetTrainer):

paddlex/modules/{3d_bev_detection → m_3d_bev_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 BEVFusionTrainer
 from .dataset_checker import BEVFusionDatasetChecker
 from .evaluator import BEVFusionEvaluator
 from .exportor import BEVFusionExportor
+from .trainer import BEVFusionTrainer

paddlex/modules/{3d_bev_detection → m_3d_bev_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,12 +12,12 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from pathlib import Path
 import pickle
+from pathlib import Path
 
 from ...base import BaseDatasetChecker
-from .dataset_src import check, deep_analyse
 from ..model_list import MODELS
+from .dataset_src import check, deep_analyse
 
 
 class BEVFusionDatasetChecker(BaseDatasetChecker):

paddlex/modules/{3d_bev_detection → m_3d_bev_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,5 +13,5 @@
 # limitations under the License.
 
 
-from .check_dataset import check
 from .analyse_dataset import deep_analyse
+from .check_dataset import check

paddlex/modules/{3d_bev_detection → m_3d_bev_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,24 +14,24 @@
 
 
 import os
-import json
-import platform
 import pickle
-
+import platform
 from collections import defaultdict
 
 import numpy as np
-import matplotlib.pyplot as plt
-from matplotlib import font_manager
-from matplotlib.backends.backend_agg import FigureCanvasAgg
 
+from paddlex.utils.deps import function_requires_deps, is_dep_available
 from paddlex.utils.fonts import PINGFANG_FONT_FILE_PATH
 
+if is_dep_available("matplotlib"):
+    import matplotlib.pyplot as plt
+    from matplotlib import font_manager
+
 
+@function_requires_deps("matplotlib")
 def deep_analyse(dataset_dir, output):
     """class analysis for dataset"""
     tags = ["train", "val"]
-    all_instances = 0
     class_name_train = defaultdict(int)
     class_name_val = defaultdict(int)
     for tag in tags:

paddlex/modules/{3d_bev_detection → m_3d_bev_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.
@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import os
 import os.path as osp
 import pickle
 

paddlex/modules/{3d_bev_detection → m_3d_bev_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.
@@ -23,7 +23,7 @@ class BEVFusionEvaluator(BaseEvaluator):
     entities = MODELS
 
     def update_config(self):
-        """update evalution config"""
+        """update evaluation config"""
         if self.eval_config.batch_size is not None:
             self.pdx_config.update_batch_size(self.eval_config.batch_size)
         self.pdx_config.update_dataset(
@@ -35,7 +35,7 @@ class BEVFusionEvaluator(BaseEvaluator):
         self.pdx_config.update_weights(self.eval_config.weight_path)
 
     def get_eval_kwargs(self) -> dict:
-        """get key-value arguments of model evalution function
+        """get key-value arguments of model evaluation function
 
         Returns:
             dict: the arguments of evaluation function.

paddlex/modules/{3d_bev_detection → m_3d_bev_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/{3d_bev_detection → m_3d_bev_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.

paddlex/modules/{3d_bev_detection → m_3d_bev_detection}/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.
@@ -13,11 +13,7 @@
 # limitations under the License.
 
 
-from pathlib import Path
-
 from ..base import BaseTrainer
-from ...utils.config import AttrDict
-from ...utils import logging
 from .model_list import MODELS
 
 
@@ -51,9 +47,9 @@ class BEVFusionTrainer(BaseTrainer):
             self.pdx_config.update_learning_rate(self.train_config.learning_rate)
         if self.train_config.epochs_iters is not None:
             self.pdx_config.update_epochs(self.train_config.epochs_iters)
-            epochs_iters = self.train_config.epochs_iters
+            self.train_config.epochs_iters
         else:
-            epochs_iters = self.pdx_config.get_epochs_iters()
+            self.pdx_config.get_epochs_iters()
         if self.global_config.output is not None:
             self.pdx_config.update_save_dir(self.global_config.output)
 
@@ -65,6 +61,8 @@ class BEVFusionTrainer(BaseTrainer):
         """
         train_args = {"device": self.get_device()}
         train_args["dy2st"] = self.train_config.get("dy2st", False)
+        # amp support 'O1', 'O2', 'OFF'
+        train_args["amp"] = self.train_config.get("amp", "OFF")
         if self.global_config.output is not None:
             train_args["save_dir"] = self.global_config.output
         return train_args

paddlex/modules/multilabel_classification/__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 MLClsTrainer
 from .dataset_checker import MLClsDatasetChecker
 from .evaluator import MLClsEvaluator
 from .exportor import MLClsExportor
+from .trainer import MLClsTrainer

paddlex/modules/multilabel_classification/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.
@@ -15,8 +15,8 @@
 from pathlib import Path
 
 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 MLClsDatasetChecker(BaseDatasetChecker):