PointNet 和 PointNet++ pytorch版本 复现 modelnet40

复现一下经典的 PointNet的pytorch版本
记录一下复现的过程

代码下载

git clone https://gitcode.net/mirrors/yanx27/pointnet_pointnet2_pytorch.git

环境配置

python=3.8.13
cuda=11.2
cudnn=8.1.0

pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

 pip install tqdm  -i https://pypi.tuna.tsinghua.edu.cn/simple  --timeout=120

数据集下载

要新建 data 文件夹 然后放数据进去
data/modelnet40_normal_resampled

PointNet 开始训练

python train_classification.py

PointNet 测试

python test_classification.py --log_dir 2023-04-26_21-53/

–log_dir 在 pointnet_pointnet2_pytorch/log/classification 下面的那个日期 pointnet 的

PointNet++ 开始训练

python train_classification.py --model pointnet2_cls_ssg --log_dir pointnet2_cls_ssg

PointNet++ 测试

python test_classification.py  --log_dir pointnet2_cls_ssg

不得不说一句，这个 pytorch 环境配置就是比tf方便啊

参考链接