
关于EMD的MATLAB代码分享——MSN-Point-Cloud-Completion:一种用于密集点云完成的变形与采样网络(AAAI...)
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本项目提供基于MATLAB实现的MSN-Point-Cloud-Completion,这是一种先进的神经网络模型,专门设计用于处理和改进密集点云数据,通过创新的变形与采样技术来优化3D空间中的点云表示。相关成果已在AAAI会议上发表。
MSN是一种基于学习的形状补全方法,用于密集点云完成,并能保留已知结构并生成分布均匀且密集的点云数据。在该项目中提供了一种计算点云地球移动距离(EMD)的方法实现,该算法基于拍卖机制并且仅需$O(n)$内存。
使用说明:
1. 环境和先决条件
- Pytorch 1.2.0
- CUDA 10.0
- Python 3.7
2. 编译
进入emd目录,运行`python3 setup.py install`
同样地,在expansion_penalty 和 MDS 目录中分别执行上述命令来安装扩展模块。
3. 数据和模型下载
鉴于数据集规模较大,不提供训练点云。如果需要生成自己的训练样本,请参考相应的文档说明。
4. 训练或验证
使用`python3 val.py`进行模型评估或者使用 `python3 train.py`从头开始训练新模型。
全部评论 (0)
还没有任何评论哟~


