软件工程应用与实践(7)可视化-kitti数据集

2021SC@SDUSC

open3d  

KITTI数据集由德国卡尔斯鲁厄理工学院和丰田美国技术研究院联合创办,是目前国际上最大的自动驾驶场景下的计算机视觉算法评测数据集。该数据集用于评测立体图像(stereo),光流(optical flow),视觉测距(visual odometry),3D物体检测(object detection)和3D跟踪(tracking)等计算机视觉技术在车载环境下的性能。

kitti数据集的中含有bin,rgb,label,calib这四类文件。其中中calib和label为txt文件。 我们需要逐行读入,然后存入到数组中,然后按照数组的索引获取具体的值读入到后续程序中。 

 Kitti_Dataset类 及代码分析

import os
import numpy as np
from data.calib import Calib
from data.object3d import Object3d
import cv2

class Kitti_Dataset:
    def __init__(self, dir_path, split="training"):
        super(Kitti_Dataset, self).__init__()
        self.dir_path = os.path.join(dir_path, split)
        # calib矫正参数文件夹地址
        self.calib = os.path.join(self.dir_path, "calib")
        # RGB图像的文件夹地址
        self.images = os.path.join(self.dir_path, "img")
        # 点云图像文件夹地址
        self.pcs = os.path.j
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
KITTI数据集可视化可以使用Mayavi库进行操作。可以根据以下代码对KITTI数据集进行可视化: ```python import mayavi.mlab as mlab dataset = kitti_object(os.path.join(ROOT_DIR, '../dataset/KITTI/object')) data_idx = 10 # Load data from dataset objects = dataset.get_label_objects(data_idx) print("There are %d objects.", len(objects)) img = dataset.get_image(data_idx) img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) img_height, img_width, img_channel = img.shape pc_velo = dataset.get_lidar(data_idx)[:,0:3] calib = dataset.get_calibration(data_idx) # Visualization using Mayavi mlab.figure(bgcolor=(0, 0, 0), size=(img_width, img_height)) mlab.points3d(pc_velo[:, 0], pc_velo[:, 1], pc_velo[:, 2], color=(1, 1, 1), mode='point') mlab.imshow(img, figure=mlab.gcf()) mlab.show() ``` 这段代码会根据给定的数据索引,加载对应的图像和点云数据,并使用Mayavi进行可视化展示。点云数据会以白色点的形式显示,而图像则会作为背景显示。这样可以直观地查看KITTI数据集中的图像和对应的点云信息。请确保在代码中设置了正确的数据集路径。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *2* *3* [KITTI数据集可视化(一):点云多种视图的可视化实现](https://blog.csdn.net/weixin_44751294/article/details/127345052)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT3_1"}}] [.reference_item style="max-width: 100%"] [ .reference_list ]

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值