最近参照服务器运行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
再次运行,没有问题