nuScenes数据集在OpenPCDet中的使用及其获取

烤粽子

发布于 2021-10-08 09:17:46

6K00

代码可运行

文章被收录于专栏：粽子的深度学习笔记粽子的深度学习笔记

运行总次数：0

代码可运行

1. 安装官方提供的开发者工具

pip install nuscenes-devkit==1.0.5

2. 下载数据

从官方网站上下载数据NuScenes 3D object detection dataset，没注册的需要注册后下载。 注意： 如果觉得数据下载或者创建data infos有难度的，可以参考本文下方 5.

3. 数据组织结构

下载好数据集后按照文件结构解压放置。其在OpenPCDet中的数据结构及其位置如下，根据自己使用的数据是v1.0-trainval，还是v1.0-mini来修改。

OpenPCDet
├── data
│   ├── nuscenes
│   │   │── v1.0-trainval (or v1.0-mini if you use mini)
│   │   │   │── samples
│   │   │   │── sweeps
│   │   │   │── maps
│   │   │   │── v1.0-trainval  
├── pcdet
├── tools

4. 创建data infos

根据数据选择

python -m pcdet.datasets.nuscenes.nuscenes_dataset --func create_nuscenes_infos \
    --cfg_file tools/cfgs/dataset_configs/nuscenes_dataset.yaml \
    --version v1.0-mini

或者

python -m pcdet.datasets.nuscenes.nuscenes_dataset --func create_nuscenes_infos \
    --cfg_file tools/cfgs/dataset_configs/nuscenes_dataset.yaml \
    --version v1.0-trainval

5. 数据获取新途径

如果觉得数据下载或者创建data infos有难度的，可以考虑使用本人处理好的数据 v1.0-mini v1.0-trainval 数据待更新… 其主要存放的结构为

│── v1.0-trainval (or v1.0-mini if you use mini)
│   │── gt_database_10sweeps_withvelo
│   │── nuscenes_infos_10sweeps_train.pkl
│   │── nuscenes_infos_10sweeps_val.pkl
│   │── nuscenes_dbinfos_10sweeps_withvelo.pkl

6. train

bash scripts/dist_train.sh 8 --cfg_file cfgs/nuscenes_models/cbgs_pp_multihead.yaml

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2021/09/30 ，如有侵权请联系 cloudcommunity@tencent.com 删除

data

dataset

detection

object

本文分享自作者个人站点/博客前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体同步曝光计划，欢迎热爱写作的你一起参与！

登录后参与评论

0 条评论

热度

nuScenes数据集在OpenPCDet中的使用及其获取

nuScenes数据集在OpenPCDet中的使用及其获取

1. 安装官方提供的开发者工具

2. 下载数据

3. 数据组织结构

4. 创建data infos

5. 数据获取新途径

6. train

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐