问题原因
代码目录结构如下:
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
改成你自己的路径即可
然后训练过程如下: