---
title: GenCAD-model
canonical_url: "https://www.modelscope.cn/models/wax2wane/GenCAD-model"
md_url: "https://www.modelscope.cn/models/wax2wane/GenCAD-model.md"
repository: wax2wane/GenCAD-model
chinese_name: "方便下载模型git clone https://github.com/ferdous-alam/GenCAD"
last_updated: 2026-05-19
downloads: 4
stars: 1
---

# GenCAD-model

> GenCAD-model - wax2wane 在 ModelScope 开源的模型。复制网盘，git clone https://github.com/ferdous-alam/GenCAD。预训练模型需要从 Google Drive 下载：- 链接： https://drive.google.com/drive/folders/1Ej7wdtlqT5P-SoUf3gsZXD8b78XqhiI5

- **Repository**: wax2wane/GenCAD-model
- **Downloads**: 4
- **Stars**: 1
- **Last updated**: 2026-05-19

Source: https://www.modelscope.cn/models/wax2wane/GenCAD-model

---

# 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')"
```
### 推荐顺序
1. 先运行 python inference_gencad.py 验证基本环境
2. 如果推理成功，可以尝试训练测试
3. 如果遇到问题，检查上述环境配置


<br />

<br />