|
@@ -19,12 +19,6 @@ from importlib import import_module
|
|
|
import lazy_paddle
|
|
import lazy_paddle
|
|
|
|
|
|
|
|
from ....utils import logging
|
|
from ....utils import logging
|
|
|
-
|
|
|
|
|
-if lazy_paddle.is_compiled_with_cuda() and not lazy_paddle.is_compiled_with_rocm():
|
|
|
|
|
- from ....ops.voxelize import hard_voxelize
|
|
|
|
|
- from ....ops.iou3d_nms import nms_gpu
|
|
|
|
|
-else:
|
|
|
|
|
- logging.error("3D BEVFusion custom ops only support GPU platform!")
|
|
|
|
|
from ....utils.func_register import FuncRegister
|
|
from ....utils.func_register import FuncRegister
|
|
|
|
|
|
|
|
module_3d_bev_detection = import_module(".3d_bev_detection", "paddlex.modules")
|
|
module_3d_bev_detection = import_module(".3d_bev_detection", "paddlex.modules")
|
|
@@ -92,6 +86,12 @@ class BEVDet3DPredictor(BasicPredictor):
|
|
|
Returns:
|
|
Returns:
|
|
|
tuple: A tuple containing the preprocessors and inference engine.
|
|
tuple: A tuple containing the preprocessors and inference engine.
|
|
|
"""
|
|
"""
|
|
|
|
|
+ if lazy_paddle.is_compiled_with_cuda() and not lazy_paddle.is_compiled_with_rocm():
|
|
|
|
|
+ from ....ops.voxelize import hard_voxelize
|
|
|
|
|
+ from ....ops.iou3d_nms import nms_gpu
|
|
|
|
|
+ else:
|
|
|
|
|
+ logging.error("3D BEVFusion custom ops only support GPU platform!")
|
|
|
|
|
+
|
|
|
pre_tfs = {"Read": ReadNuscenesData()}
|
|
pre_tfs = {"Read": ReadNuscenesData()}
|
|
|
for cfg in self.config["PreProcess"]["transform_ops"]:
|
|
for cfg in self.config["PreProcess"]["transform_ops"]:
|
|
|
tf_key = list(cfg.keys())[0]
|
|
tf_key = list(cfg.keys())[0]
|