基于nuScenes数据集的pointpillars代码复现(second.pytorch)

nuScenes数据集加速下载请参阅如下博客:

https://blog.csdn.net/weixin_38423729/article/details/111059557

second.pytorch代码地址如下:

https://github.com/traveller59/second.pytorch

具体操作内容见源码,本文章旨在疏通运行前的细节

创建数据集命令,必须加上--dataset_name="NuScenesDataset".

python create_data.py nuscenes_data_prep --root_path=/home/user/nuScenes-v1.0-trainval --version="v1.0-trainval" --max_sweeps=10 --dataset_name="NuScenesDataset"
python create_data.py nuscenes_data_prep --data_path=NUSCENES_TEST_DATASET_ROOT --version="v1.0-test" --max_sweeps=10 --dataset_name="NuScenesDataset"

由于nuscenes-devkit版本更新,在使用create_data.py创建数据集时代码会出现"cvpr_2019"无效的错误,需要修改两部分

1.cvpr_2019修改为detection_cvpr_2019

2.将注释部分修改为未注释部分

# from nuscenes.eval.detection.config import eval_detection_configs 
# cls_range_map = eval_detection_configs[eval_version]["class_range"]
from nuscenes.eval.detection.config import config_factory
from nuscenes.eval.detection.config import DetectionConfig
cfg = config_factory(eval_version)
cls_range_map = cfg.class_range

未完待续。。。

  • 1
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值