# GenCAD：图片驱动的 CAD 生成框架

> 基于 Transformer 对比表示与扩散先验，输入草图或图片即可生成 3D CAD 模型，TMCE 2025 论文

## 项目简介
GenCAD 是一个图片条件化的计算机辅助设计（CAD）生成框架，基于 Transformer 对比表示（Contrastive Representation）与扩散先验（Diffusion Priors），能够将草图或实物图片转换为可编辑的 3D CAD 模型。

## 核心技术
| 模块 | 说明 |
|------|------|
| CSR Model | Contrastive Sketch Representation，对比学习草图与 CAD 的表示 |
| CCIP Model | Cross-Conditional Image Prior，跨模态图像先验 |
| Diffusion Prior | 扩散模型生成 CAD 嵌入，实现高质量形状生成 |

## 功能特点
- 输入草图→输出 CAD 模型
- 输入实物图片→输出 CAD 模型
- 支持 STL 可视化转换
- 支持无头服务器（headless）运行

## 部署方式
```bash
# Docker 推荐
git clone https://github.com/ferdous-alam/GenCAD
cd GenCAD
docker build -t gencad:latest .

# 训练 CSR Model
docker run -it gencad:latest conda run -n gencad_env python train_gencad.py csr -name test -gpu 0

# 推理
docker run --gpus all -v $(pwd)/data/images:/app/data/images -v $(pwd)/results:/app/results -it gencad:latest /bin/bash
xvfb-run --server-args="-screen 0 2048x2048x24" python inference_gencad.py -image_path data/images -export_img
```

## 论文与资料
- Paper: TMCE 2025
- arXiv: 2409.16254

## 适用场景
- 工业设计自动化：将手绘草图或实物照片转为可编辑 CAD
- AI 辅助造型：快速生成 3D 模型原型
- 计算机图形学研究：CAD 生成、跨模态表示学习

## Stars
2.1k Stars，Python 实现，活跃维护中

---

**分类**：源码
**标签**：模型 · 生成 · CAD
**作者**：Xiao.Xi
**链接**：https://octohz.com/p/1697