pointnet.pytorch 复现_balls.py

作者：菜鸟追梦旅行 | 2024-05-12 00:05:51

踩

balls.py

@[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
1
2

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

进入python环境，输入

import torch
torch.backends.cudnn.version()
1
2

#查看是否是gpu版本

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

训练 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("…/")

在这里插入图片描述
顺序可能有点问题，为了方便自己看的, 先这样了咯，拜了个拜～

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/菜鸟追梦旅行/article/detail/556772