pytorch_coma

在火炬中重新实现COMA 这是Pytorch中代码的重新实现。 如果使用此代码，请遵循作者的许可权。 要求 此代码已在Python3.6，Pytorch versoin 1.3上进行了测试。 可以通过运行安装要求 pip install -r requirements.txt 从安装网格处理库。 注意，为此需要网状软件包库的python3版本。 火车 要开始培训，请按照以下步骤操作 从下载注册数据 通过运行生成默认配置文件。 python config_parser.py 将以下参数添加到default.cfg data_dir-步骤1中下载的数据集的路径（例如/ home / pixelite1201 / mesh_raw /） checkpoint_dir-将存储检查点的路径 visual_output_dir-如果将visualize设置为True，则评估的视觉输出将存