PointNet-Pytorch代码运行(分类部分)

1、下载源码及环境的准备

1.1源码下载

PointNet 源码下载地址:https://github.com/fxia22/pointnet.pytorch

分类数据集为modelnet40

数据集下载官网(我从官网下载失败了,然后去其他地方下载的)

Princeton ModelNeticon-default.png?t=N6B9http://modelnet.cs.princeton.edu/

1.2环境准备

安装Pointnet包

cd pointnet.pytorch-master
pip install -e .

2、代码运行

cd utils
python train_classification.py --dataset --nepoch= --dataset_type (运行分类代码)
python train_segmentation.py --dataset --nepoch= (运行分割代码)

--dataset=后面为你的数据路径

--nepoch=5,你的epoch的次数,我设置的5

--dataset_type=shapenet,设置数据的类型,我的是shapenet,若为modelnet40,就设置为--dataset_type=modelnet40

报错1 importError: No module named ‘pointnet‘

http://t.csdn.cn/Q2aKL

报错2 not found trainval.txt

这个源码就是少了这几个文件!!可以在源码作者的issues的#52获得缺少的txt文件

报错3 not found //.ply

这个代码读取的是ply格式数据集,要将数据集转成ply格式

报错4:The "freeze_support()" line can be omitted if the program is not going to be frozen to produce an executable.

参考:http://t.csdn.cn/8Ku8Q

碎碎念:本人代码小白,这个代码运行过程中出现很多报错

参考内容:

(四) 三维点云课程---PointNet-Pytorch运行_pointnet代码运行_月夕花晨TS的博客-CSDN博客

Pointnet代码运行及数据集的制作_颉傲的博客-CSDN博客

pointnet-pytorch运行步骤_ThetaQing的博客-CSDN博客


 

你可能感兴趣的:(pytorch,人工智能,python)