Skip to content

Commit 75ad85c

Browse files
committed
Include original instance and semantic label in evaluation point cloud
1 parent d6f7155 commit 75ad85c

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

src/ros/ros_utils.cpp

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -322,7 +322,7 @@ sensor_msgs::PointCloud2Ptr evaluationToPointCloud(const std::vector<KittiSegmen
322322
msg->is_dense = false;
323323

324324
sensor_msgs::PointCloud2Modifier output_modifier(*msg);
325-
output_modifier.setPointCloud2Fields(8,
325+
output_modifier.setPointCloud2Fields(10,
326326
"x",
327327
1,
328328
sensor_msgs::PointField::FLOAT32,
@@ -332,6 +332,12 @@ sensor_msgs::PointCloud2Ptr evaluationToPointCloud(const std::vector<KittiSegmen
332332
"z",
333333
1,
334334
sensor_msgs::PointField::FLOAT32,
335+
"semantic_label",
336+
1,
337+
sensor_msgs::PointField::UINT16,
338+
"instance_label",
339+
1,
340+
sensor_msgs::PointField::UINT16,
335341
"has_corresponding_point_in_detection_point_cloud",
336342
1,
337343
sensor_msgs::PointField::UINT8,
@@ -351,6 +357,8 @@ sensor_msgs::PointCloud2Ptr evaluationToPointCloud(const std::vector<KittiSegmen
351357
sensor_msgs::PointCloud2Iterator<float> iter_x_out(*msg, "x");
352358
sensor_msgs::PointCloud2Iterator<float> iter_y_out(*msg, "y");
353359
sensor_msgs::PointCloud2Iterator<float> iter_z_out(*msg, "z");
360+
sensor_msgs::PointCloud2Iterator<uint16_t> iter_semantic_label_out(*msg, "semantic_label");
361+
sensor_msgs::PointCloud2Iterator<uint16_t> iter_instance_label_out(*msg, "instance_label");
354362
sensor_msgs::PointCloud2Iterator<uint8_t> iter_has_corresponding_point_in_detection_point_cloud_out(
355363
*msg, "has_corresponding_point_in_detection_point_cloud");
356364
sensor_msgs::PointCloud2Iterator<uint32_t> iter_gpe_out(*msg, "ground_point_evaluation");
@@ -374,6 +382,8 @@ sensor_msgs::PointCloud2Ptr evaluationToPointCloud(const std::vector<KittiSegmen
374382
*(iter_x_out + i) = p.point.x;
375383
*(iter_y_out + i) = p.point.y;
376384
*(iter_z_out + i) = p.point.z;
385+
*(iter_semantic_label_out + i) = p.point.semantic_label;
386+
*(iter_instance_label_out + i) = p.point.instance_label;
377387
*(iter_has_corresponding_point_in_detection_point_cloud_out + i) =
378388
p.has_corresponding_point_in_detection_point_cloud ? WHITE : RED;
379389
*(iter_gpe_out + i) = ground_point_evaluation_color;

0 commit comments

Comments
 (0)