在tensorflow2.0环境下使用RandLA-Net训练S3DIS数据集

之前的文章介绍了在tensorflow2.0环境下使用RandLA-Net训练Semantic3D数据集,这里我们记录一下如何在在tensorflow2.0环境下使用RandLA-Net训练S3DIS数据集。

由于作者提供的代码是基于tensorflow1.11写的,想在2.0的环境下运行该程序需要做些修改,修改方法已在训练Semantic3D数据集中给出。

训练S3DIS数据集

1、下载数据集
作者使用的是Stanford3dDataset_v1.2,整个压缩包下载下来是4.79G,解压后有30个G。

2、数据集预处理
数据集下载后解压缩,原始数据中存在一定错误,Area_5\office_19\Annotations\ceiling_1.txt中有一行数据中包含了字符,导致数据操作失败,详细过程参见使用NumPy加载txt文件提示ValueError:could not convert string to float

然后执行下面语句

python data_prepare_s3dis.py

3、6折交叉训练
先修改自己的数据路径,然后执行下面代码

sh jobs_6_fold_cv_s3dis.sh

4、验证
将test文件夹下所有的ply文件全部拷贝到/data/S3DIS/res

  • 3
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 36
    评论
RandLA-Net是一种高效的点云分割模型,它可以处理大规模点云数据,且具有优秀的性能。如果想要使用RandLA-Net训练自己的数据集,需要以下几个步骤: 1. 数据准备:首先需要准备自己的点云数据集,可以使用lidar或者其他传感器采集。需要保证数据集的标注准确、齐全。如果没有标注,可以使用现有的工具进行手动或者半自动标注。 2. 数据预处理:将采集到的数据进行处理,包括点云去噪、下采样、法向量估计等操作。同时还需要将数据划分为训练集、验证集和测试集。 3. 模型训练使用RandLA-Net对数据集进行训练。可以使用现有的预训练模型对数据集进行微调,或者从头开始训练。在训练过程中,需要设置好损失函数、优化器等参数,并且需要注意过拟合的问题。 4. 模型验证:训练好的模型需要进行验证,以确保其性能符合要求。可以使用验证集进行测试,并且需要评估模型在准确率、召回率、F1值等方面的性能。 5. 模型优化:如果模型的性能不够优秀,可以进行模型优化。例如调整模型的超参数、修改损失函数等,以达到更好的性能。 总的来说,使用RandLA-Net训练自己的数据集需要进行数据准备、预处理、模型训练、模型验证和模型优化等多个步骤。需要根据具体数据集的情况进行参数调整,以达到更好的效果。
评论 36
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

GHZhao_GIS_RS

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值