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

paddlex/modules/ts_forecast/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,13 +13,11 @@
 # limitations under the License.
 
 import os
-import json
-import time
 import tarfile
 from pathlib import Path
 
+from ...utils.flags import FLAGS_json_format_model
 from ..base import BaseTrainer
-from ...utils.config import AttrDict
 from .model_list import MODELS
 
 
@@ -35,9 +33,9 @@ class TSFCTrainer(BaseTrainer):
         self.update_config()
         self.dump_config()
         train_args = self.get_train_kwargs()
-        export_with_pir = self.global_config.get("export_with_pir", False) or os.getenv(
-            "FLAGS_json_format_model"
-        ) in ["1", "True"]
+        export_with_pir = (
+            self.global_config.get("export_with_pir", False) or FLAGS_json_format_model
+        )
         train_args.update(
             {
                 "uniform_output_enabled": self.train_config.get(
@@ -105,4 +103,6 @@ training!"
         train_args = {"device": self.get_device(using_device_number=1)}
         if self.global_config.output is not None:
             train_args["save_dir"] = self.global_config.output
+        # amp support 'O1', 'O2', 'OFF'
+        train_args["amp"] = self.train_config.get("amp", "OFF")
         return train_args

paddlex/modules/video_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 VideoClsTrainer
 from .dataset_checker import VideoClsDatasetChecker
 from .evaluator import VideoClsEvaluator
 from .exportor import VideoClsExportor
+from .trainer import VideoClsTrainer

paddlex/modules/video_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, split_dataset, deep_analyse
 from ..model_list import MODELS
+from .dataset_src import check, deep_analyse, split_dataset
 
 
 class VideoClsDatasetChecker(BaseDatasetChecker):

paddlex/modules/video_classification/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,6 +13,6 @@
 # limitations under the License.
 
 
+from .analyse_dataset import deep_analyse
 from .check_dataset import check
 from .split_dataset import split_dataset
-from .analyse_dataset import deep_analyse

paddlex/modules/video_classification/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.
@@ -13,22 +13,22 @@
 # limitations under the License.
 
 import os
-import json
-import math
 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 .....utils.deps import function_requires_deps, is_dep_available
+
+if is_dep_available("matplotlib"):
+    import matplotlib.pyplot as plt
+    from matplotlib import font_manager
 
 from .....utils.file_interface import custom_open
 from .....utils.fonts import PINGFANG_FONT_FILE_PATH
 
 
+@function_requires_deps("matplotlib")
 def deep_analyse(dataset_path, output):
     """class analysis for dataset"""
     tags = ["train", "val"]

paddlex/modules/video_classification/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,10 +15,9 @@
 import os
 import os.path as osp
 import random
-from PIL import Image, ImageOps
 from collections import defaultdict
 
-from .....utils.errors import DatasetFileNotFoundError, CheckFailedError
+from .....utils.errors import CheckFailedError, DatasetFileNotFoundError
 
 
 def check(dataset_dir, output, sample_num=10):

paddlex/modules/video_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/video_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.
@@ -22,7 +22,7 @@ class VideoClsEvaluator(BaseEvaluator):
     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(
@@ -33,7 +33,7 @@ class VideoClsEvaluator(BaseEvaluator):
         self.pdx_config.update_pretrained_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/video_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/video_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/video_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 VideoClsTrainer(BaseTrainer):
@@ -85,4 +83,6 @@ class VideoClsTrainer(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/video_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 VideoDetTrainer
 from .dataset_checker import VideoDetDatasetChecker
 from .evaluator import VideoDetEvaluator
 from .exportor import VideoDetExportor
+from .trainer import VideoDetTrainer

paddlex/modules/video_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.
@@ -15,8 +15,8 @@
 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 VideoDetDatasetChecker(BaseDatasetChecker):

paddlex/modules/video_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/video_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.
@@ -13,22 +13,21 @@
 # limitations under the License.
 
 import os
-import json
-import math
 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 .....utils.deps import function_requires_deps, is_dep_available
 from .....utils.file_interface import custom_open
 from .....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_path, output):
     """class analysis for dataset"""
     tags = ["train", "val"]

paddlex/modules/video_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.
@@ -15,11 +15,10 @@
 import os
 import os.path as osp
 import random
-from PIL import Image, ImageOps
 from collections import defaultdict
 
+from .....utils.errors import CheckFailedError, DatasetFileNotFoundError
 from .....utils.file_interface import custom_open
-from .....utils.errors import DatasetFileNotFoundError, CheckFailedError
 
 
 def check(dataset_dir, output, sample_num=10):
@@ -30,7 +29,6 @@ def check(dataset_dir, output, sample_num=10):
         raise DatasetFileNotFoundError(file_path=dataset_dir)
 
     tags = ["train", "val"]
-    delim = " "
     valid_num_parts = 5
 
     sample_cnts = dict()
@@ -104,7 +102,7 @@ def check(dataset_dir, output, sample_num=10):
                         for label_line in label_lines:
                             label_info = label_line.strip().split(" ")
                             try:
-                                label = int(label_info[0])
+                                int(label_info[0])
                             except (ValueError, TypeError) as e:
                                 raise CheckFailedError(
                                     f"Ensure that the first number in each line in {label_info} should be int."

paddlex/modules/video_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,7 +22,7 @@ class VideoDetEvaluator(BaseEvaluator):
     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(
@@ -31,7 +31,7 @@ class VideoDetEvaluator(BaseEvaluator):
         self.pdx_config.update_pretrained_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/video_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/video_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/video_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.
@@ -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 VideoDetTrainer(BaseTrainer):
@@ -79,4 +77,6 @@ class VideoDetTrainer(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/ops/__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.
@@ -19,8 +19,9 @@ import sys
 from types import ModuleType
 
 import filelock
-from paddle.utils.cpp_extension import load as paddle_jit_load
+
 from ..utils import logging
+from ..utils.deps import class_requires_deps
 
 
 def get_user_home() -> str:
@@ -97,6 +98,8 @@ class PaddleXCustomOperatorModule(ModuleType):
         super().__init__(modulename)
 
     def jit_build(self):
+        from paddle.utils.cpp_extension import load as paddle_jit_load
+
         try:
             lockfile = "paddlex.ops.{}".format(self.modulename)
             lockfile = os.path.join(TMP_HOME, lockfile)
@@ -111,7 +114,7 @@ class PaddleXCustomOperatorModule(ModuleType):
             with filelock.FileLock(lockfile):
                 return paddle_jit_load(name=self.modulename, sources=sources, **args)
         except:
-            logging.error("{} builded fail!".format(self.modulename))
+            logging.error("{} built fail!".format(self.modulename))
             raise
 
     def _load_module(self):
@@ -123,7 +126,7 @@ class PaddleXCustomOperatorModule(ModuleType):
                     "No custom op {} found, try JIT build".format(self.modulename)
                 )
                 self.module = self.jit_build()
-                logging.info("{} builded success!".format(self.modulename))
+                logging.info("{} built success!".format(self.modulename))
 
             # refresh
             sys.modules[self.fullname] = self.module

paddlex/ops/iou3d_nms/iou3d_cpu.cpp

@@ -76,7 +76,7 @@ inline int check_in_box2d(const float *box, const Point &p) {
   float center_x = box[0], center_y = box[1];
   float angle_cos = cos(-box[6]),
         angle_sin =
-            sin(-box[6]);  // rotate the point in the opposite direction of box
+            sin(-box[6]); // rotate the point in the opposite direction of box
   float rot_x = (p.x - center_x) * angle_cos + (p.y - center_y) * (-angle_sin);
   float rot_y = (p.x - center_x) * angle_sin + (p.y - center_y) * angle_cos;
 
@@ -87,7 +87,8 @@ inline int check_in_box2d(const float *box, const Point &p) {
 inline int intersection(const Point &p1, const Point &p0, const Point &q1,
                         const Point &q0, Point &ans) {
   // fast exclusion
-  if (check_rect_cross(p0, p1, q0, q1) == 0) return 0;
+  if (check_rect_cross(p0, p1, q0, q1) == 0)
+    return 0;
 
   // check cross standing
   float s1 = cross(q0, p1, p0);
@@ -95,7 +96,8 @@ inline int intersection(const Point &p1, const Point &p0, const Point &q1,
   float s3 = cross(p0, q1, q0);
   float s4 = cross(q1, p1, q0);
 
-  if (!(s1 * s2 > 0 && s3 * s4 > 0)) return 0;
+  if (!(s1 * s2 > 0 && s3 * s4 > 0))
+    return 0;
 
   // calculate intersection of two lines
   float s5 = cross(q1, p1, p0);
@@ -238,9 +240,9 @@ inline float iou_bev(const float *box_a, const float *box_b) {
   return s_overlap / fmaxf(sa + sb - s_overlap, EPS);
 }
 
-std::vector<paddle::Tensor> boxes_iou_bev_cpu(
-    const paddle::Tensor &boxes_a_tensor,
-    const paddle::Tensor &boxes_b_tensor) {
+std::vector<paddle::Tensor>
+boxes_iou_bev_cpu(const paddle::Tensor &boxes_a_tensor,
+                  const paddle::Tensor &boxes_b_tensor) {
   // params boxes_a_tensor: (N, 7) [x, y, z, dx, dy, dz, heading]
   // params boxes_b_tensor: (M, 7) [x, y, z, dx, dy, dz, heading]
   // params ans_iou_tensor: (N, M)

paddlex/ops/iou3d_nms/iou3d_cpu.h

@@ -21,7 +21,8 @@
 
 #include <vector>
 
-std::vector<paddle::Tensor> boxes_iou_bev_cpu(
-    const paddle::Tensor& boxes_a_tensor, const paddle::Tensor& boxes_b_tensor);
+std::vector<paddle::Tensor>
+boxes_iou_bev_cpu(const paddle::Tensor &boxes_a_tensor,
+                  const paddle::Tensor &boxes_b_tensor);
 
 #endif

paddlex/ops/iou3d_nms/iou3d_nms.cpp

@@ -41,8 +41,9 @@ void NmsLauncher(const cudaStream_t &stream, const float *boxes, int64_t *mask,
 void NmsNormalLauncher(const cudaStream_t &stream, const float *boxes,
                        int64_t *mask, int boxes_num, float nms_overlap_thresh);
 
-std::vector<paddle::Tensor> boxes_overlap_bev_gpu(
-    const paddle::Tensor &boxes_a, const paddle::Tensor &boxes_b) {
+std::vector<paddle::Tensor>
+boxes_overlap_bev_gpu(const paddle::Tensor &boxes_a,
+                      const paddle::Tensor &boxes_b) {
   // params boxes_a: (N, 7) [x, y, z, dx, dy, dz, heading]
   // params boxes_b: (M, 7) [x, y, z, dx, dy, dz, heading]
   // params ans_overlap: (N, M)
@@ -61,9 +62,9 @@ std::vector<paddle::Tensor> boxes_overlap_bev_gpu(
   return {ans_overlap};
 }
 
-std::vector<paddle::Tensor> boxes_iou_bev_gpu(
-    const paddle::Tensor &boxes_a_tensor,
-    const paddle::Tensor &boxes_b_tensor) {
+std::vector<paddle::Tensor>
+boxes_iou_bev_gpu(const paddle::Tensor &boxes_a_tensor,
+                  const paddle::Tensor &boxes_b_tensor) {
   // params boxes_a: (N, 7) [x, y, z, dx, dy, dz, heading]
   // params boxes_b: (M, 7) [x, y, z, dx, dy, dz, heading]
   // params ans_overlap: (N, M)
@@ -135,7 +136,8 @@ std::vector<paddle::Tensor> nms_gpu(const paddle::Tensor &boxes,
   }
 
   num_to_keep_data[0] = num_to_keep;
-  if (cudaSuccess != cudaGetLastError()) printf("Error!\n");
+  if (cudaSuccess != cudaGetLastError())
+    printf("Error!\n");
 
   return {keep, num_to_keep_tensor};
 }

paddlex/ops/iou3d_nms/iou3d_nms.h

@@ -21,10 +21,12 @@
 
 #include <vector>
 
-std::vector<paddle::Tensor> boxes_overlap_bev_gpu(
-    const paddle::Tensor &boxes_a, const paddle::Tensor &boxes_b);
-std::vector<paddle::Tensor> boxes_iou_bev_gpu(
-    const paddle::Tensor &boxes_a_tensor, const paddle::Tensor &boxes_b_tensor);
+std::vector<paddle::Tensor>
+boxes_overlap_bev_gpu(const paddle::Tensor &boxes_a,
+                      const paddle::Tensor &boxes_b);
+std::vector<paddle::Tensor>
+boxes_iou_bev_gpu(const paddle::Tensor &boxes_a_tensor,
+                  const paddle::Tensor &boxes_b_tensor);
 std::vector<paddle::Tensor> nms_gpu(const paddle::Tensor &boxes,
                                     float nms_overlap_thresh);
 std::vector<paddle::Tensor> nms_normal_gpu(const paddle::Tensor &boxes,

paddlex/ops/iou3d_nms/iou3d_nms_api.cpp

@@ -21,13 +21,15 @@
 #include "iou3d_cpu.h"
 #include "iou3d_nms.h"
 
-std::vector<paddle::DataType> BoxesIouBevCpuInferDtype(
-    paddle::DataType boxes_a_dtype, paddle::DataType boxes_b_dtype) {
+std::vector<paddle::DataType>
+BoxesIouBevCpuInferDtype(paddle::DataType boxes_a_dtype,
+                         paddle::DataType boxes_b_dtype) {
   return {boxes_a_dtype};
 }
 
-std::vector<std::vector<int64_t>> BoxesIouBevCpuInferShape(
-    std::vector<int64_t> boxes_a_shape, std::vector<int64_t> boxes_b_shape) {
+std::vector<std::vector<int64_t>>
+BoxesIouBevCpuInferShape(std::vector<int64_t> boxes_a_shape,
+                         std::vector<int64_t> boxes_b_shape) {
   return {{boxes_a_shape[0], boxes_b_shape[0]}};
 }
 
@@ -35,38 +37,42 @@ std::vector<paddle::DataType> NmsInferDtype(paddle::DataType boxes_dtype) {
   return {paddle::DataType::INT64, paddle::DataType::INT64};
 }
 
-std::vector<std::vector<int64_t>> NmsInferShape(
-    std::vector<int64_t> boxes_shape) {
+std::vector<std::vector<int64_t>>
+NmsInferShape(std::vector<int64_t> boxes_shape) {
   return {{boxes_shape[0]}, {1}};
 }
 
-std::vector<paddle::DataType> NmsNormalInferDtype(
-    paddle::DataType boxes_dtype) {
+std::vector<paddle::DataType>
+NmsNormalInferDtype(paddle::DataType boxes_dtype) {
   return {paddle::DataType::INT64, paddle::DataType::INT64};
 }
 
-std::vector<std::vector<int64_t>> NmsNormalInferShape(
-    std::vector<int64_t> boxes_shape) {
+std::vector<std::vector<int64_t>>
+NmsNormalInferShape(std::vector<int64_t> boxes_shape) {
   return {{boxes_shape[0]}, {1}};
 }
 
-std::vector<paddle::DataType> BoxesIouBevGpuInferDtype(
-    paddle::DataType boxes_a_dtype, paddle::DataType boxes_b_dtype) {
+std::vector<paddle::DataType>
+BoxesIouBevGpuInferDtype(paddle::DataType boxes_a_dtype,
+                         paddle::DataType boxes_b_dtype) {
   return {boxes_a_dtype};
 }
 
-std::vector<std::vector<int64_t>> BoxesIouBevGpuInferShape(
-    std::vector<int64_t> boxes_a_shape, std::vector<int64_t> boxes_b_shape) {
+std::vector<std::vector<int64_t>>
+BoxesIouBevGpuInferShape(std::vector<int64_t> boxes_a_shape,
+                         std::vector<int64_t> boxes_b_shape) {
   return {{boxes_a_shape[0], boxes_b_shape[0]}};
 }
 
-std::vector<paddle::DataType> BoxesOverlapBevGpuInferDtype(
-    paddle::DataType boxes_a_dtype, paddle::DataType boxes_b_dtype) {
+std::vector<paddle::DataType>
+BoxesOverlapBevGpuInferDtype(paddle::DataType boxes_a_dtype,
+                             paddle::DataType boxes_b_dtype) {
   return {boxes_a_dtype};
 }
 
-std::vector<std::vector<int64_t>> BoxesOverlapBevGpuInferShape(
-    std::vector<int64_t> boxes_a_shape, std::vector<int64_t> boxes_b_shape) {
+std::vector<std::vector<int64_t>>
+BoxesOverlapBevGpuInferShape(std::vector<int64_t> boxes_a_shape,
+                             std::vector<int64_t> boxes_b_shape) {
   return {{boxes_a_shape[0], boxes_b_shape[0]}};
 }