pointnet.pytorch在win10下实现过程

参考博客;基于Pytorch训练Pointnet+Windows10 - Nadir-Echo - 博客园 (cnblogs.com)

1.安装anaconda环境

2.创建一个虚拟环境，其中的torch版本不符合的可以按照官网给出配置

链接;Start Locally | PyTorch

3.需要按照参考博客修改代码，使用的是分割数据集。应该跑分割部分代码，segment部分，分类也能运行，不过结果不对。

路径部分需要自行修改

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument(
        '--batchSize', type=int, default=2, help='input batch size')
    parser.add_argument(
        '--workers', type=int, help='number of data loading workers', default=4)
    parser.add_argument(
        '--nepoch', type=int, default=5, help='number of epochs to train for')
    parser.add_argument('--outf', type=str, default='seg', help='output folder')
    parser.add_argument('--model', type=str, default='', help='model path')
    parser.add_argument('--dataset', type=str,default='E:\\pcdlcode\\pointnet.pytorch\\shapenetcore_partanno_segmentation_benchmark_v0\\', help="dataset path")
    parser.add_argument('--class_choice', type=str, default='Chair', help="class_choice")
    parser.add_argument('--feature_transform', action='store_true', help="use feature transform")

    opt = parser.parse_args()
    print(opt)

4.训练， batchsize训练不可设置为1

代码含有Batch Normalization

不能设置1

 由于本人电脑配置有限，只能跑通，没跑完。

还有很多一知半解，不懂得地方，小弟哪里写的不对，请各位大神多多指正。