这里主要以S3DIS数据集为例来解释一下pointnet的源码
博客点云语义分割:PointNet训练S3DIS数据集记录了如何使用pointnet训练S3DIS模型。直接执行train.py就可以了。这里我们就记录一下train.py的每一步操作。
大致可以分为参数设置、数据加载、训练参数设置、单轮训练、单轮验证几个过程。
一、参数设置
参数设置中主要是指定测试集是哪个区域:- -test_area,然后是一些模型训练的超参数,如batch_size,学习率、衰减率、优化器等
parser = argparse.ArgumentParser()
parser.add_argument('--gp