PointPillars复现问题小结

项目地址:https://github.com/traveller59/second.pytorch
由于pointpillar是在second的基础上改动的,所以直接用second项目,只不过在训练和测试时参数文件不一样。
1、
首先安装pytorch,使用anaconda新建一个环境,百度搜索pytorch,从官网选择cuda,直接有安装命令,即可安装pytorch。
pytorch支持cuda10.1,所以首先要安装cuda10.1
2、
然后按照项目安装依赖,这里要注意,
llvmlite要选择0.26.0版本,使用conda安装 conda install llvmlite==0.26.0
3、
安装spconv,注意安装高版本cmake,注意cmake安装方法在其readme中有,有三个指令,要分别运行,加sudo
add second.pytorch/ to PYTHONPATH 这一步指的是在bashrc文件中添加项目地址,可以百度
注意修改配置文件中的文件目录,
./ 表示当前文件夹
…/表示上一级文件夹

注意训练时,由于GPU显存不够,batchsize要调整为1,多进程可以调为8

网络还有一个小bug
/pytorch/aten/src/ATen/native/cuda/LegacyDefinitions.cpp:14: UserWarning: masked_fill_ received a mask with dtype torch.uint8, this behavior is now deprecated,please use a mask with dtype torch.bool instead.
要把progress_bar.py中的Unit.Byte 改为Unit.Bool
也有人说是
在second.pytorch\second\pytorch\models文件夹下的voxelnet.py的911行:
改为:
opp_labels = (box_preds[…, -1] > 0) ^ dir_labels.bool()
就可以正常训练了(本版本没有找到这个在哪)

The same problem. Fixed it by changing every
“mask = blahblahblah.byte()”
to
“mask = blahblahblah.bool()”

训练完后用tensorboard查看训练过程:
tensorboard --logdir=logs
logs为日志文件夹,具体本例为summary文件夹,注意需要安装tensorflow才能使用

要进行BBox可视化,还需要安装一些扩展
Ros-kinetic-jsk-gui-msgs
-hark-msgs
-interactive
-interactive-marker
-interactive-test
-recognition-msgs
-recognition-utils
-rqt-plugins
-rviz-plugins
-topic-tools
-visualization

注意:运行ros时说没有rospkg :

  1. $ conda install setuptools
  2. $ pip install -U rosdep rosinstall_generator wstool rosinstall six vcstools

来自 https://blog.csdn.net/qq_36501182/article/details/79971570

由于可视化用到了nuscenes,所以要安装nuscenes:
Pip install nuscenes-devkit

numpy版本不能大于1.18
Pip uninstall numpy
Pip install numpy==1.17
Num_worker数越多,占用内存越大,但是好像和时间没有关系
batchsize越大,每个step用时越多,某一step的训练量越大

RTX显卡可以使用apex
安装apex时出错,ninja有问题,把-v 改成 --version,还是有问题,在梯度削减的地方出错。。。放弃后面再试
https://blog.csdn.net/Orientliu96/article/details/104583998

KITTI 3D目标检测评估参数说明
https://blog.csdn.net/shuqiaos/article/details/82770370

  • 0
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
复现OpenPCDet中的PointPillars算法,可以按照以下步骤进行操作: 1. 首先,克隆OpenPCDet的GitHub仓库到本地。可以使用以下命令进行克隆: ``` git clone https://github.com/open-mmlab/OpenPCDet.git ``` 2. 进入克隆好的OpenPCDet目录,并安装所需的Python依赖。可以使用以下命令安装依赖: ``` cd OpenPCDet pip install -r requirements.txt ``` 3. 下载PointPillars的源代码和相关环境包。可以使用以下命令进行下载: ``` mkdir -p ~/pointpillars_ros/src cd ~/pointpillars_ros/src git clone https://github.com/BIT-DYN/pointpillars_ros ``` 4. 将OpenPCDet工程中的OpenPCDet/tools目录下的全部文件以及/OpenPCDet/pcdet目录下的文件复制到~/pointpillars_ros/src/pointpillars_ros/tools目录下。确保将所有文件都复制过去,尤其要注意不要遗漏pcdet文件夹。 5. 进入~/pointpillars_ros目录,并激活OpenPCDet的环境。可以使用以下命令进行激活: ``` cd ~/pointpillars_ros conda activate pcdet ``` 6. 安装所需的ROS依赖。可以使用以下命令进行安装: ``` pip install --user rospkg catkin_pkg sudo apt-get install ros-noetic-pcl-ros sudo apt-get install ros-noetic-jsk-recognition-msg sudo apt-get install ros-noetic-jsk-rviz-plugins ``` 7. 完成上述步骤后,即可开始复现PointPillars算法。可以使用以下命令启动PointPillars: ``` source ~/pointpillars_ros/devel/setup.bash roslaunch pointpillars_ros pointpillars.launch ``` 请注意,以上步骤仅为大致指导,具体操作可能因环境和配置而有所不同。建议参考OpenPCDet和PointPillars的官方文档以获取更详细的指导。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值