custom-diffusion360

用几张图定制扩散模型，还能自由控制相机视角生成新图

CustomDiffusion360 是一个基于 Stable Diffusion 的文本生成图像定制化扩展项目，核心解决现有定制化模型（如 DreamBooth、Custom Diffusion）只能生成固定视角、无法控制相机视角的问题。它通过引入相机姿态控制，让用户仅用少量参考图像（如3-5张）即可微调扩散模型，生成同一物体或风格在不同视角下的新图像，并支持显式指定相机位置（如俯视、侧视、旋转）。项目基于 PyTorch 实现，提供完整的训练和推理流程，支持自定义数据集和姿态条件注入，适用于三维感知的图像生成、虚拟商品展示、创意设计等场景。其核心能力包括：少样本定制化、视角可控生成、与现有文本到图像模型兼容。作为早期项目，代码结构清晰，但文档和社区生态尚在建设中。