denoising_diffusion

用PyTorch Lightning复现DDPM,快速跑通扩散模型训练与生成

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 图像生成 🍴 复刻 12 🐛 待处理 2 👀 关注 2
GitHub ↗
GitHub 星标 ★ 91
维护状态 低维护
是否开源 开源
定价模式 🏷️完全免费 开源项目,MIT许可证,完全免费,可自行部署使用。

项目预览

denoising_diffusion 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类图像生成
开发团队hmdolatabadi
定价模式🏷️完全免费
价格说明开源项目,MIT许可证,完全免费,可自行部署使用。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议MIT
主要语言Python
技术栈/模型denoising-diffusion,generative-modeling,image-generation,pytorch-lightning
GitHub 星标★ 91
上线时间2020-08-30 00:00:00
代码更新2020-10-04
仓库体积21.5 MB
本站数据更新2026-11-10
维护状态低维护
收录时间2026-08-09
浏览次数3

项目介绍

这是一个非官方的PyTorch实现,复现了论文《Denoising Diffusion Probabilistic Models》(DDPM)中的去噪扩散概率模型。该项目旨在帮助研究者和开发者快速理解并运行扩散模型,解决生成式图像建模中的训练稳定性和生成质量等问题。核心能力包括:基于PyTorch Lightning框架实现简洁的训练流程,支持自定义数据集和图像尺寸,提供完整的训练、采样和评估脚本,以及清晰的代码结构便于二次开发。项目代码量精简,适合作为学习扩散模型原理的参考实现,也可作为实验基线进行修改和扩展。

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 91

核心亮点

  • 代码结构清晰,基于PyTorch Lightning简化训练流程,易于上手
  • 完整实现DDPM前向加噪和反向去噪过程,适合学习原理
  • 支持自定义数据集和图像尺寸,灵活适配不同实验需求

不足之处

  • 项目为早期版本,文档和示例较少,社区支持待观察
  • 未提供预训练模型,需自行训练才能评估生成效果

适用场景

  • 学习扩散模型原理和PyTorch实现细节
  • 快速搭建扩散模型训练实验基线
  • 在自定义图像数据集上验证DDPM生成效果

替代项目

openai/guided-diffusion、lucidrains/denoising-diffusion-pytorch、huggingface/diffusers

同类项目推荐

oumi 开源

一条命令搞定 LLM 微调、评估与部署,让模型落地更简单。

Easily fine-tune (SFT/RL), evaluate, and deploy Qwen, Gemma, or any open agentic LLM···

★ 9429 2026-08-10
data-forcing-distillation 开源

少步生成视频,不丢多样性和画质

Data-Forcing Distillation (DFD): restoring diversity and fidelity in few-step video ···

★ 65 2026-08-09
MyLLM 开源

跟着 Notebook 从数据到应用跑通大模型全流程

"LLM from Zero to Hero: An End-to-End Large Language Model Journey from Data to Appl···

★ 153 2026-09-15
Visual-ERM 开源

训练视觉奖励模型,让AI输出更符合人类视觉偏好

[NeurIPS 2026] Official Implementation of "Visual-ERM: Reward Modeling for Visual Eq···

★ 68 2026-08-09
CUDA-L2 开源

用强化学习自动写出比 cuBLAS 更快的矩阵乘法内核

CUDA-L2: Surpassing cuBLAS Performance for Matrix Multiplication through Reinforceme···

★ 487 2026-08-09
ringattention 开源

环形分布注意力,让长上下文处理又快又省显存

Large Context Attention

★ 775 2026-08-09
ETO 开源

让 LLM Agent 在试错中自我进化,越用越聪明

Trial and Error: Exploration-Based Trajectory Optimization of LLM Agents (ACL 2024 M···

★ 170 2026-08-09
amazon-sagemaker-examples 开源

照着跑一遍,轻松上手 AWS 机器学习全流程

Example Jupyter notebooks that demonstrate how to build, train, and deploy machine ···

★ 11003 2026-08-09
Adam-mini 开源

用更少学习率,训更快更大模型,省显存提性能。

Code for Adam-mini: Use Fewer Learning Rates To Gain More https://arxiv.org/abs/2406···

★ 462 2026-08-09