chore: add ml detectors' buffer size (#1067)

Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]>
autowarefoundation · Jul 17, 2024 · 29686a6 · 29686a6
1 parent 2602ebd
commit 29686a6
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 0 deletions.
diff --git a/..._launch/config/perception/object_recognition/detection/lidar_model/centerpoint.param.yaml b/..._launch/config/perception/object_recognition/detection/lidar_model/centerpoint.param.yaml
@@ -7,6 +7,7 @@
     head_onnx_path: "$(var model_path)/pts_backbone_neck_head_$(var model_name).onnx"
     head_engine_path: "$(var model_path)/pts_backbone_neck_head_$(var model_name).engine"
     trt_precision: fp16
+    cloud_capacity: 2000000
     post_process_params:
       # post-process params
       circle_nms_dist_threshold: 0.5

diff --git a/...h/config/perception/object_recognition/detection/lidar_model/centerpoint_sigma.param.yaml b/...h/config/perception/object_recognition/detection/lidar_model/centerpoint_sigma.param.yaml
@@ -17,6 +17,7 @@
     has_variance: true
     has_twist: true
     trt_precision: fp16
+    cloud_capacity: 2000000
     densification_num_past_frames: 1
     densification_world_frame_id: map
 

diff --git a/...ch/config/perception/object_recognition/detection/lidar_model/centerpoint_tiny.param.yaml b/...ch/config/perception/object_recognition/detection/lidar_model/centerpoint_tiny.param.yaml
@@ -7,6 +7,7 @@
     head_onnx_path: "$(var model_path)/pts_backbone_neck_head_$(var model_name).onnx"
     head_engine_path: "$(var model_path)/pts_backbone_neck_head_$(var model_name).engine"
     trt_precision: fp16
+    cloud_capacity: 2000000
     post_process_params:
       # post-process params
       circle_nms_dist_threshold: 0.5

diff --git a/...aunch/config/perception/object_recognition/detection/lidar_model/pointpainting.param.yaml b/...aunch/config/perception/object_recognition/detection/lidar_model/pointpainting.param.yaml
@@ -5,6 +5,7 @@
     head_onnx_path: "$(var model_path)/pts_backbone_neck_head_$(var model_name).onnx"
     head_engine_path: "$(var model_path)/pts_backbone_neck_head_$(var model_name).engine"
     trt_precision: fp16
+    cloud_capacity: 2000000
     post_process_params:
       # post-process params
       circle_nms_dist_threshold: 0.3