基于 AIGC 的三维场景生成系统

项目简介

面向 AIGC 3D 原型验证，搭建模块化生成系统，串联“文生图 → 深度图 → 点云 → 网格”流程。系统支持可控文生图、参数消融与自动报告，并提供 CLI 和 Gradio 交互式 Demo。

核心实现

• 可控图像生成：基于 Diffusers 和 Stable Diffusion v1.5 构建参数化模块，支持 CFG、steps、scheduler、negative prompt、seed 和 multi-sample 配置。
• 实验与结果对比：支持参数化多样本生成、可复现实验、批量对比和自动消融，输出 CSV 记录及自动报告推荐。
• 三维生成流程：结合 Depth Anything V2、Open3D 和 TripoSR，组织深度估计、点云与网格生成模块。
• 输入预处理：针对复杂背景增加去背景与前景比例调节策略，提高 Image-to-3D 生成的稳定性。
• 交互演示：通过 CLI 和 Gradio 提供生成入口，结合 Plotly 展示结果。

技术栈

Python、PyTorch、Diffusers、Stable Diffusion v1.5、Depth Anything V2、Open3D、TripoSR、Gradio、Plotly。