当模型训练稳定之后,就开始对模型进行部署了,RandLANet的dataset中包含了训练、验证和测试的所有过程,实际部署时我们只需要做推理即可,同时RandLANet的输出结果只是一个label标签,我们希望能够直接输出带有坐标的分割结果。
一、数据预处理
在main_Semantic.py的基础上修改,加入一个新的函数preprocess_pc
,也就是把预处理的过程加进来,同时还加入离群点去除的操作,将两次下采样修改为只做一次下采样。
from os.path import join, exists,dirname,abspath
from RandLANet import Network
fro