PyTorch中的MeshCNN
SIGGRAPH 2019
MeshCNN是用于3D三角形网格的通用深度神经网络,可用于诸如3D形状分类或分割之类的任务。 该框架包括直接应用于网格边缘的卷积,池化和解池层。
该代码由和在支持下编写。
入门
安装
克隆此仓库:
git clone https://github.com/ranahanocka/MeshCNN.git
cd MeshCNN
安装依赖项: 1.2版。 可选: 用于训练图。 通过新的conda环境conda env create -f environment.yml (创建一个名为meshcnn的环境)
SHREC上的3D形状分类
下载数据集
bash ./scripts/shrec/get_data.sh
运行训练(如果使用conda env首先激活env,例如source activate meshcnn )
bash ./scripts/shrec/train.sh
要查看训练损失图,请在另一个终端中运行tensorboard --logdir runs并单击 。
运行测试并导出中间池网格:
bas
1