BAAF-Net训练Semantic3D数据集
BAAF-Net的作者在github上只放出了S3DIS数据集的训练和测试代码,而且我也问了作者近期并没有放出关于Semantic3D数据集的训练和测试代码的计划,但是作者说他是基于RandLA-Net的代码做的,可以参考RandLA-Net训练Semantic3D数据集的代码进行修改,刚好我们已经对Randla-Net的代码非常熟悉了。在这篇文章里我们就尝试使用BAAF-Net训练Semantic3D数据集。
![](https://i-blog.csdnimg.cn/blog_migrate/65b87439b5971279ab44c70b0de57dcb.png)
一、数据预处理
参考在tensorflow2.0环境下使用RandLA-Net训练Semantic3D数据集中Semantic3D的数据预处理过程。
二、训练和测试代码修改
将RandLaNet的main_semantic.py脚本拷贝到BAAF-Net目录下。两处修改
1、网络引用
from RandLANet import Network
改为