ROS学习番外篇7—将SemanticKITTI数据集的点云投影到图像平面

做自动驾驶的老哥相信或多或少会知道SemanticKITTI这个点级精细标注的数据集。该数据集基于kitti的odometry数据集进行标注。因此非但有精细的语言分割信息,而且还能结合kitti数据集本身就提供的四个摄像头信息来做更多的事情。比如将点云投影到图像上,然后做融合算法。
在这里插入图片描述
在这篇文章中我们将介绍将点云投影到图像平面的操作。

数据下载

首先,我们需要先下载图像数据,也就是到kitti的官网下载相应的数据:
在这里插入图片描述
如果先前已经下了SemanticKITTI数据集,那么就只需要下图像的数据就行。

坐标投影

在这里插入图片描述
kitti数据集的坐标和转换关系如上图所示,把点云投影到图像上需要先将其坐标转换到参考坐标系中,然后再将其投影到摄像头对应的坐标系。在odometry数据集中,这个过程涉及到两个矩阵,全部在calib文件中给出中给出,下面我们将直接通过代码展示将点云投影到左边彩色相机的操作:


                
  • 0
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
要将KITTI数据集点云数据投影到2D图像,可以使用ROS中的点云转换工具包`pcl_ros`来实现。具体步骤如下: 1. 安装`pcl_ros`工具包 ```bash sudo apt-get install ros-<distro>-pcl-ros ``` 其中`<distro>`是你使用的ROS版本,比如`melodic`。 2. 下载KITTI数据集KITTI官网下载点云数据集。 3. 创建ROS包并将数据集导入 在ROS工作空间中创建一个ROS包,将KITTI数据集导入到ROS包中。 4. 编写ROS节点 使用`pcl_ros`中的`PointCloud2`消息类型读取点云数据,并使用`sensor_msgs/Image`消息类型发布投影后的2D图像。 具体代码可以参考以下示例: ```python import rospy import sensor_msgs.point_cloud2 as pc2 from sensor_msgs.msg import Image, CameraInfo from cv_bridge import CvBridge import numpy as np import cv2 class PointCloudProjector(): def __init__(self): self.cloud_sub = rospy.Subscriber("/kitti/velo/pointcloud", pc2.PointCloud2, self.cloud_callback) self.image_pub = rospy.Publisher("/kitti/velo/image_raw", Image, queue_size=10) self.cam_info_pub = rospy.Publisher("/kitti/velo/camera_info", CameraInfo, queue_size=10) self.bridge = CvBridge() def cloud_callback(self, msg): # Convert PointCloud2 message to numpy array cloud = pc2.read_points(msg, skip_nans=True) # Project point cloud onto 2D image # ... # Publish image image_msg = self.bridge.cv2_to_imgmsg(image, encoding="passthrough") self.image_pub.publish(image_msg) # Publish camera info cam_info_msg = CameraInfo() # ... self.cam_info_pub.publish(cam_info_msg) if __name__ == '__main__': rospy.init_node('point_cloud_projector', anonymous=True) projector = PointCloudProjector() rospy.spin() ``` 在`cloud_callback`函数中,使用`pc2.read_points`函数将`PointCloud2`消息转换为numpy数组,然后将点云投影到2D图像中,并将结果转换为ROS中的`Image`消息类型,使用`self.image_pub.publish`函数发布2D图像。 同时,还需要发布相机信息,可使用`CameraInfo`消息类型,使用`self.cam_info_pub.publish`函数发布。 注意,这里的投影算法需要根据具体需求进行选择和实现。 5. 运行ROS节点 使用`rosrun`命令运行ROS节点。 ```bash rosrun <your_package_name> point_cloud_projector.py ``` 其中,`<your_package_name>`是你创建的ROS包的名称。 6. 查看投影结果 使用`rviz`或其他图像查看工具查看2D图像
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值