简介:MeshCNN是在PyTorch框架下实现的一种用于处理三维网格数据的卷积神经网络方法,专为非规则结构设计。
MeshCNN是SIGGRAPH 2019上提出的一种用于3D三角形网格的通用深度神经网络框架,适用于诸如3D形状分类或分割的任务。该框架包括直接应用于网格边缘的卷积、池化和解池层等操作。
安装指南如下:
- 克隆此仓库
```bash
git clone https://github.com/ranahanocka/MeshCNN.git
cd MeshCNN
```
- 安装依赖项:使用1.2版。通过新的conda环境`conda env create -f environment.yml`(创建一个名为meshcnn的环境)。
对于SHREC上的3D形状分类任务,具体步骤如下:
- 下载数据集
```bash
bash .scripts/shrec/get_data.sh
```
- 运行训练:在激活相应conda环境后执行脚本。例如,在终端中输入`source activate meshcnn`后再运行以下命令。
```bash
bash .scripts/shrec/train.sh
```
查看训练损失图,可以在另一个终端中通过tensorboard --logdir runs并单击相关链接来完成。
- 运行测试及导出中间池网格:
```bash
bas