MeshAnythingV2 项目使用教程

MeshAnythingV2 项目使用教程

MeshAnythingV2 From anything to mesh like human artists. Official impl. of "MeshAnything V2: Artist-Created Mesh Generation With Adjacent Mesh Tokenization" 项目地址: https://gitcode.com/gh_mirrors/me/MeshAnythingV2

1. 项目目录结构及介绍

MeshAnythingV2 项目目录结构如下：

MeshAnythingV2/
├── demo/              # 示例文件夹，包含示例模型
├── examples/          # 示例输入文件夹，包含用于测试的模型文件
├── gt_examples/       # 地标示例文件夹，包含用于比较的模型文件
├── pc_examples/       # 点云示例文件夹，包含点云数据文件
├── app.py             # 本地Gradio演示的启动文件
├── adjacent_mesh_tokenization.py # 相邻网格符号化的实现文件
├── main.py            # 主程序文件，用于模型的训练和推理
├── mesh_to_pc.py      # 网格转点云的脚本文件
├── requirements.txt   # 项目依赖文件
├── LICENSE.txt        # 项目许可证文件
└── README.md          # 项目说明文件

• demo/: 包含项目演示的一些示例文件。
• examples/: 包含用于模型推理的输入示例模型文件。
• gt_examples/: 包含用于比较的地标示例模型文件。
• pc_examples/: 包含点云数据文件，用于点云到网格的转换。
• app.py: 用于启动本地Gradio演示的Python脚本。
• adjacent_mesh_tokenization.py: 实现相邻网格符号化的Python脚本。
• main.py: 项目的主脚本，用于执行模型的训练和推理。
• mesh_to_pc.py: 将网格数据转换为点云数据的脚本。
• requirements.txt: 包含项目依赖的Python包列表。
• LICENSE.txt: 项目的开源许可证信息。
• README.md: 项目的详细说明文档。

2. 项目的启动文件介绍

app.py 是用于启动本地Gradio演示的文件。运行以下命令可以启动演示：

python app.py

这将启动一个本地服务器，并通过Web界面提供交互式的模型演示。

main.py 是项目的主启动文件，用于执行模型的训练和推理操作。以下是一些使用 main.py 的示例命令：

• 使用Marching Cubes预处理输入网格并生成艺术家风格的网格：

python main.py --input_dir examples --out_dir mesh_output --input_type mesh --mc

• 对于单个文件输入，使用以下命令：

python main.py --input_path examples/wand.obj --out_dir mesh_output --input_type mesh --mc

• 对于点云输入，使用以下命令：

python main.py --input_dir pc_examples --out_dir pc_output --input_type pc_normal

3. 项目的配置文件介绍

本项目主要依赖于环境配置和Python包的安装，没有专门的配置文件。确保项目能够正常运行，需要按照以下步骤设置环境：

1. 克隆项目仓库：

git clone https://github.com/buaacyw/MeshAnythingV2.git && cd MeshAnythingV2

2. 创建并激活conda环境：

conda create -n MeshAnythingV2 python==3.10.13 -y
conda activate MeshAnythingV2

3. 安装必要的Python包：

pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118
pip install -r requirements.txt
pip install flash-attn --no-build-isolation
pip install -U gradio

完成以上步骤后，项目环境即配置完成，可以开始运行项目中的脚本进行模型的训练或推理。

MeshAnythingV2 From anything to mesh like human artists. Official impl. of "MeshAnything V2: Artist-Created Mesh Generation With Adjacent Mesh Tokenization" 项目地址: https://gitcode.com/gh_mirrors/me/MeshAnythingV2