在OpenPCDet训练KITTI

1. 数据集准备工作

(1)按照官方给出的OpenPCDet中的KITTI数据集的组织方法,排布数据集

(2)然后运行下面的命令生成数据info

python -m pcdet.datasets.kitti.kitti_dataset create_kitti_infos tools/cfgs/dataset_configs/kitti_dataset.yaml

(3)会在kitti目录下面生成5个pkl文件 :

2. 训练数据

(1)单卡训练

python train.py --cfg_file cfgs/kitti_models/pv_rcnn.yaml

(2)多卡训练

CUDA_VISIBLE_DEVICES=3,6,7 python -m torch.distributed.launch --nproc_per_node=3 train.py --cfg_file cfgs/kitti_models/pv_rcnn.yaml --launcher pytorch

用几张卡--nproc_per_node的值就去取多少,防止溢出

注:史少帅博士的介绍:https://zhuanlan.zhihu.com/p/152120636

代码理解:https://blog.csdn.net/weixin_44128857/article/details/108516213

  • 4
    点赞
  • 25
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值