自动驾驶数据集waymo

目录

waymo-open-dataset-viewer

依赖项：

可视化参考： 

---

waymo-open-dataset-viewer

https://github.com/erksch/waymo-open-dataset-viewer/tree/master

依赖项：

# 安装 tensorflow （需要注意tensorflow与cuda版本对应，我这里使用cuda 11.0版本）
pip install tensorflow==2.4.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

#安装 pytorch (pytorch-lightning 是否安装自行选择）
pip install pytorch-lightning==1.4.0
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

# 测试gpu是否可用（返回 gpu 信息和 True 则表示安装正确）
python
>>> import tensorflow as tf
>>> tf.config.list_physical_devices('GPU')
[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]
>>> import torch
>>> torch.cuda.is_available()
True
>>> quit()

 # 安装 waymo-open-dataset 需要与tensorflow版本对应
 pip install waymo-open-dataset-tf-2-4-0

waymo-open-dataset 没找到，pypi有安装包，只支持linux 系统。

可视化参考： 

https://blog.csdn.net/weixin_50232758/article/details/132260047

 

• 数据集官网：https://waymo.com/open/
• 数据集开发工具包：https://github.com/waymo-research/waymo-open-dataset
  （依靠数据集开发工具包，可以了解标签的更多信息，并可以使用 Python 阅读标签，可视化点云）
• 官方下载地址：https://waymo.com/open/download/

https://github.com/waymo-research/waymo-open-dataset/tree/master/docs