Skip to content

Commit b049286

Browse files
committed
improve[FoundationPose]: Change default value of min_depth & max_depth
Signed-off-by: zz990099 <771647586@qq.com>
1 parent 45f221c commit b049286

2 files changed

Lines changed: 3 additions & 3 deletions

File tree

detection_6d_foundationpose/src/foundationpose.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ class FoundationPose : public Base6DofDetectionModel {
3333
const int max_input_image_W = 1920,
3434
const int crop_window_H = 160,
3535
const int crop_window_W = 160,
36-
const float min_depth = 0.1);
36+
const float min_depth = 0.001);
3737

3838
bool Register(const cv::Mat &rgb,
3939
const cv::Mat &depth,
@@ -295,7 +295,7 @@ bool FoundationPose::UploadDataToDevice(
295295
convert_depth_to_xyz_map(static_cast<float *>(depth_on_device), input_image_height,
296296
input_image_width, static_cast<float *>(xyz_map_on_device),
297297
intrinsic_(0, 0), intrinsic_(1, 1), intrinsic_(0, 2), intrinsic_(1, 2),
298-
0.1);
298+
0.001);
299299

300300
// 输出device端指针,并注册析构过程
301301
auto func_release_cuda_buffer = [](void *ptr) {

detection_6d_foundationpose/src/foundationpose_render.hpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ class FoundationPoseRenderer {
2323
const int input_poses_num,
2424
const int crop_window_H = 160,
2525
const int crop_window_W = 160,
26-
const float min_depth = 0.1,
26+
const float min_depth = 0.001,
2727
const float max_depth = 4.0);
2828

2929
bool RenderAndTransform(const std::vector<Eigen::Matrix4f> &_poses,

0 commit comments

Comments
 (0)