将.pcd文件可视化为鸟瞰图(俯视图)

首先,你可以用pcl库中的 pcl_viewer 3D可视化 .pcd 文件。
在做3D检测的时候可能需要用到鸟瞰图来检查效果,所以写了这么个功能:
这里需要注意的 .pcd 不能为二进制文件,否则不能直接读取。

import numpy as np
import cv2

def lidar_to_bird_view_img(lidar, factor=4):
    # Input:
    #   lidar: (N', 4)
    #   factor: 越大清晰度越高
    # Output:
    #   birdview: (w, l, 3)
    birdview = np.zeros(
        (INPUT_HEIGHT * factor, INPUT_WIDTH * factor, 1))
    for point in lidar:
        x, y = point[0:2]
        if X_MIN < x < X_MAX and Y_MIN < y < Y_MAX:
            x, y = int((x - X_MIN) / VOXEL_X_SIZE *
                       factor), int((y - Y_MIN) / VOXEL_Y_SIZE * factor)
            birdview[y, x] += 1
    birdview = birdview - np.min(birdview)
    divisor = np.max(birdview) - np.min(birdview)
    # TODO: adjust this factor
    birdview = np.clip((birdview / divisor * 255) *
                       5 * factor, a_min=0, a_max=255)
    birdview = np.tile(birdview, 3).astype(np.uint8)

    return birdview

# 限定可视化范围
INPUT_HEIGHT = 400
INPUT_WIDTH = 352
X_MIN = 0
X_MAX = 70.4
Y_MIN = -40
Y_MAX = 40
VOXEL_X_SIZE = 0.2
VOXEL_Y_SIZE = 0.2


lidar = []
with open('000000.pcd','r') as f:
    line = f.readline().strip()
    while line:
        linestr = line.split(" ")
        if len(linestr) == 4:
        	# 将str类型转换为float类型
            linestr_convert = list(map(float, linestr))
            lidar.append(linestr_convert)
        line = f.readline().strip()

bird_view  = lidar_to_bird_view_img(lidar)
bird_view_path = '000000.jpg'
cv2.imwrite( bird_view_path, bird_view )

显示效果如下:
在这里插入图片描述

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值