-
成功在Pytorch中实现3DSSD的项目:3DSSD-pytorch-openPCDet
资源介绍
3DSSD-pytorch-openPCDet
在Pytorch中成功实施3DSSD
感谢 ! 3DSSD的此实现主要基于pcdet v0.3。
准备
克隆此存储库
安装Python依赖项。
pip install -r requirements.txt
选项:安装spconv。 请按照的说明进行。 此3DSSD不使用spconv。 但是,为了更好地使用OpenPCDet,我建议安装它。
安装pcdet库。
python setup.py develop
安装pointnet2_3DSSD库。
cd pcdet/ops/pointnet2/pointnet2_3DSSD/
python setup.py develop
训练模型
我已将默认配置文件设置为3DSSD模型。 因此,您可以运行:
cd tools
python train.py
训练日志和张量板日志保存到输出目
- 上一篇: IEC62368-1-2018版标准.pdf
- 下一篇: Detector:pytorch检测器