pointnet.pytorch 复现

@[TOC] linux下 Pointnet pytorh 版本
NVIDIA-SMI 460.56 Driver Version: 460.56 CUDA Version: 11.2 复现时产生的问题以及解决方法

conda install pytorch=1.7.1 cudatoolkit=11.0 -c pytorch

https://www.pythonheidong.com/blog/article/182191/ccb0fb20d587dc92648a/
https://www.pythonheidong.com/blog/article/182191/ccb0fb20d587dc92648a/

python show_cls.py --model cls/cls_model_0.pth

运行show_cls.py

python show_cls.py --model cls/cls_model_0.pth

运行show_seg.py

python show_seg.py --model seg/seg_model_Chair_3.pth --class_choice Airplane

查看已经安装的Pytorch对应的CUDA版本

进入python环境，输入

import torch
torch.version.cuda

查看已经安装的Pytorch对应的CUDNN版本

进入python环境，输入

import torch
torch.backends.cudnn.version()

#查看是否是gpu版本

flag = torch.cuda.is_available()  
print(flag)

训练 train_classification.py

python train_classification.py --dataset ‘/home/wnang/pointnet.pytorch/shapenetcore_partanno_segmentation_benchmark_v0’ --nepoch=‘5’ --dataset_type ‘shapenet’

训练 train_segmentation.py

python train_segmentation.py --dataset ‘/home/wnang/pointnet.pytorch/shapenetcore_partanno_segmentation_benchmark_v0’ --nepoch=‘5’

https://www.freesion.com/article/1470918699/

(pointnet) wnang@wnang-OMEN-Laptop-15-ek0xxx:~/pointnet.pytorch/utils$ python show3d_balls.py

pip install opencv-python

no mudule named pointnet 解决方法： 在segmentation中加入以下字段

import sys
sys.path.append("…/")

顺序可能有点问题 ，为了方便自己看的, 先这样了咯，拜了个拜～