simple-diffusion

用最简代码看懂扩散模型,快速跑通图像生成

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

项目预览

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

项目数据

分类图像生成
开发团队filipbasara0
定价模式🏷️完全免费
价格说明开源项目,MIT许可证,完全免费,可自行部署使用。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议MIT
主要语言Python
技术栈/模型attention-mechanism,computer-vision,deep-learning,diffusion,image-generation,pytorch,stable-diffusion,unet
GitHub 星标★ 137
上线时间2022-10-20 00:00:00
代码更新2025-12-05
仓库体积40 KB
本站数据更新2026-11-08
维护状态低维护
收录时间2026-08-09
浏览次数3

项目介绍

simple-diffusion 是一个用 PyTorch 实现的最小化去噪扩散模型,旨在以极简代码展示扩散模型的核心原理。它解决了初学者理解扩散模型门槛高的问题,通过清晰的代码结构演示了前向加噪、反向去噪、UNet 主干、注意力机制等关键组件。项目不依赖复杂框架,适合教学和快速实验,但功能上不包含 Stable Diffusion 的文本条件生成等高级特性。核心能力包括:基础的 DDPM 训练与采样流程、UNet 网络结构、以及可扩展的注意力模块。对于想深入理解扩散模型底层机制的研究者或开发者,这是一个理想的入门参考。

Star 增长趋势

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

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

核心亮点

  • 代码极简,聚焦扩散模型核心,无冗余封装,易读易改
  • 完整实现 DDPM 训练与采样流程,可快速复现生成效果
  • 包含 UNet 和注意力机制,为后续扩展 Stable Diffusion 打基础

不足之处

  • 功能单一,不支持文本条件生成,无法直接用于实际产品
  • 文档/社区待观察

适用场景

  • 教学演示:在课堂或工作坊中讲解扩散模型原理
  • 快速原型:验证新想法时作为基线代码
  • 个人学习:通过阅读源码深入理解 DDPM 内部机制

替代项目

denoising-diffusion-pytorch、diffusers、openai/improved-diffusion

同类项目推荐

get_model 开源

模块化建模跨细胞类型转录调控,多模态数据一键训练

A modular framework for multimodal cross-cell-type transcriptional regulation models

★ 114 2026-08-09
AlignProp 开源

直接反向传播奖励,让扩散模型更听话,效率提升25倍

AlignProp uses direct reward backpropogation for the alignment of large-scale text-t···

★ 330 2026-08-09
neptune-client 开源

一键记录训练实验,自动追踪指标与模型,团队协作更高效

The experiment tracker for foundation model training

★ 622 2026-08-09
mrmr 开源

高维数据下自动选特征,去冗余提精度

mRMR (minimum-Redundancy-Maximum-Relevance) for automatic feature selection at scale···

★ 631 2026-08-09
modalities 开源

PyTorch 原生分布式训练,配置化复现大模型实验

Modalities, a PyTorch-native framework for distributed and reproducible foundation m···

★ 112 2026-09-19
PyTorch-CycleGAN 开源

用无配对图片训练,轻松实现风格迁移和图像转换

A clean and readable Pytorch implementation of CycleGAN

★ 1319 2026-08-09
makeMoE 开源

手写稀疏混合专家模型,彻底搞懂 MoE 原理

From scratch implementation of a sparse mixture of experts language model inspired b···

★ 821 2026-08-09
datablations 开源

数据不够时,教你如何最优训练大模型

Scaling Data-Constrained Language Models

★ 352 2026-08-09