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

最近在做点云语义分割,RandLA-Net是比较新的针对大规模点云的语义分割网络,之前一直使用的是torch版本的代码在训练自己的数据集,训练结果一直很差,同时我在Semantic3D数据集上训练的效果也很差,有几类精度始终为0,最终平均iou只有20%多,这和作者论文中给出的实验结果相差很大。

为了验证是否是我使用的pytorch工程代码的问题,我打算在作者自己提供的代码上测试一下Semantic3D数据集,RandLA-Net作者在github上发布的是tensorflow版本的代码,而且是tensorflow1.11,我之前训练模型一直都是torch,cuda安装的是10.2,而1.0版本的tensorflow在cuda10.2的环境下无法使用gpu(网上有人成功过,但我始终没有成功😭,然后我又不太想降低我的cuda版本,因为我的torch用的版本也比较高),就尝试着在2.0环境下训练1.0版本的代码。下面是训练过程记录。

修改方法

由于作者没有用到比较复杂的方法,所以要在tensorflow2.0环境下训练tensorflow1.0的代码只需要在导入tensorflow时做下修改即可,将工程中使用到的

import tensorflow as tf

替换为

impo
  • 2
    点赞
  • 31
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 74
    评论
在Win10系统下训练RandLA-Net点云语义分割模型,需要以下步骤: 1. 安装CUDA和CUDNN:RandLA-Net是基于深度学习框架PyTorch开发的,要在Win10系统上训练该模型,首先需要安装支持GPU计算的CUDA和CUDNN。 2. 安装Python和PyTorch:Python是深度学习常用的编程语言,安装Python后,通过pip安装PyTorch库。 3. 下载训练数据集:从相关资源网站下载点云语义分割数据集。常用的数据集Semantic3D、S3DIS等。下载后,将数据集解压至指定目录。 4. 准备训练代码和配置文件:在下载好论文源码的基础上,将代码和配置文件放置在合适的文件夹中。根据具体需求调整配置文件中的参数,比如训练epoch数、批量大小、学习率等。 5. 数据预处理:对下载的数据集进行预处理,将点云数据转换为模型可接受的输入格式。预处理过程包括点云读取、数据标准化、特征提取等步骤。 6. 开始训练:运行训练代码,开始进行模型训练训练过程中,模型将根据配置文件中的参数进行前向传播、反向传播等操作,并根据损失函数进行模型参数的更新。 7. 模型评估和测试:训练完成后,可以对模型进行评估和测试。评估过程中,使用测试集对模型进行性能评估,如计算准确率、召回率、F1分数等指标。 8. 模型应用:经过训练的模型可以用于点云语义分割任务。将未标注点云数据输入模型,进行语义分割操作,得到点云中各个点的语义标签。 需要注意的是,训练模型所需的时间和计算资源与数据集的大小和复杂度有关。较大规模的数据集和复杂的模型需要更多的时间和计算资源来训练

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 74
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

GHZhao_GIS_RS

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

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

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

打赏作者

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

抵扣说明:

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

余额充值