使用PointRCNN训练自己数据集,ModuleNotFoundError: No module named ‘pointnet2_cuda‘问题

最近参照服务器运行PointRCNN代码全过程记录_pointrcnn代码复现_今天也想干嘛呢的博客-CSDN博客

大神的文章复现pointrcnn,在执行训练自己数据集的过程中,遇见了无法找到pointnet2_cuda模块的问题:ModuleNotFoundError: No module named  pointnet2_cuda

python generate_gt_database.py --class_name 'Car' --split train
(这里发现使用软链接还是会报错,暂时没找到解决办法,目前是把数据集和代码放在一起)
 
## 一阶段
python train_rcnn.py --cfg_file cfgs/default.yaml --batch_size 16 --train_mode rpn --epochs 200
 
## 二阶段
python train_rcnn.py --cfg_file cfgs/default.yaml --batch_size 4 --train_mode rcnn --epochs 70  --ckpt_save_interval 2 --rpn_ckpt ../output/rpn/default/ckpt/checkpoint_epoch_200.pth

查阅资料找了一下pointnet2_cuda的位置代码阅读详解在Pytorch中定义自己写的CUDA编程函数 | 码农家园

按照网站里的建议,在PointRCNN/pointnet2_lib/pointnet2文件夹下运行了python setup.py install命令,同时按照提示把ball_query.cpp文件中的AT_CHECK 改为 TORCH_CHECK

再次运行,没有问题

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值