问题原因

代码目录结构如下:
在这里插入图片描述
pointnet是和utils同级的包,而utils下面的train_classification.py文件源码如下
在这里插入图片描述
它是直接import pointnet.dataset,而pointnet不在train_classification.py的同级目录下,所以无法访问。

解决办法

修改train_classification.py导入pointnet的路径即可。
train_classification.py中加入语句

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

就可以了。
运行语句
python train_classification.py --dataset 'yourFilePath/PointNet/pointnet.pytorch/shapenetcore_partanno_segmentation_benchmark_v0/' --nepoch='5' --dataset_type 'shapenet'
yourFilePath改成你自己的路径即可
然后训练过程如下:
在这里插入图片描述

更多推荐