Git项目：https://github.com/ferdous-alam/GenCAD
项目需要预训练模型才能运行推理，但目前 model/ckpt/ 目录只有一个提示文件。有两个选择：
方案一：先训练 CSR 模型（推荐用于测试）
CSR 是基础的 CAD 自编码器模型，训练它不需要预训练模型，可以直接测试环境：
git clone https://github.com/ferdous-alam/GenCAD
cd GenCAD
python train_gencad.py
这个命令会：
- 使用 data/filtered_data.json 作为训练数据
- 在 results/test/CSR/ 目录下保存训练结果和日志
- 训练完成后会生成模型 checkpoint
方案二：下载预训练模型
根据 readme，预训练模型需要从 Google Drive 下载：
- 链接： https://drive.google.com/drive/folders/1Ej7wdtlqT5P-SoUf3gsZXD8b78XqhiI5 需要下载三个文件并放到 model/ckpt/ 目录：
- ae_ckpt_epoch1000.pth
- ccip_sketch_ckpt_epoch300.pth
- sketch_cond_diffusion_ckpt_epoch1000000.pt
如果遇到问题，可以先检查环境：
# 检查 Python 版本
python --version
# 检查 PyTorch 和 CUDA
python -c "import torch; print(torch.
__version__); print(torch.cuda.
is_available())"
# 检查 pythonocc-core
python -c "from OCC.Core.BRepPrimAPI
import BRepPrimAPI_MakeBox; print
('pythonocc-core OK')"
推荐顺序
- 先运行 python inference_gencad.py 验证基本环境
- 如果推理成功，可以尝试训练测试
- 如果遇到问题，检查上述环境配置