MNISTDiffusion

从零手写扩散模型,4.5MB 轻量生成 MNIST 数字

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

项目预览

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

项目数据

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

项目介绍

MNISTDiffusion 是一个从零开始实现的去噪扩散概率模型(DDPM)的极简版本,专门用于 MNIST 手写数字生成。项目代码精简,模型大小仅 4.55MB,旨在帮助开发者快速理解扩散模型的核心原理。它解决了初学者难以从复杂代码库入手学习扩散模型的问题,提供了清晰、可读性高的 PyTorch 实现,包含训练和采样流程。核心能力包括:基于 UNet 的噪声预测网络、前向加噪与反向去噪过程、以及可视化生成结果。通过该项目,用户可以直观地看到扩散模型如何从纯噪声逐步生成逼真的数字图像,是学习扩散模型和图像生成的理想入门材料。

Star 增长趋势

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

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

核心亮点

  • 代码极简,适合初学者快速理解 DDPM 核心机制
  • 模型体积小(4.55MB),训练和推理资源需求低
  • 提供完整的训练与采样流程,开箱即用

不足之处

  • 仅支持 MNIST 数据集,未扩展到其他图像域
  • 文档/社区待观察

适用场景

  • 学习扩散模型原理的教学演示
  • 快速原型验证扩散模型生成效果
  • 作为轻量级基线对比其他生成模型

替代项目

denoising-diffusion-pytorch、Diffusion-LM、OpenAI Guided Diffusion

上一篇:caption-upsampling

下一篇:fast-stable-diffusion

同类项目推荐

amazon-bedrock-samples 开源

官方示例,快速上手 Bedrock 全家桶

This repository contains examples for customers to get started using the Amazon Bedr···

★ 1532 2026-08-09
buffer-of-thought-llm 开源

给大模型装个思想缓存,复杂推理更准更稳

[NeurIPS 2024 Spotlight] Buffer of Thoughts: Thought-Augmented Reasoning with Large ···

★ 678 2026-08-09
TokenFlow 开源

一个分词器同时搞定图像理解和生成,统一多模态。

[CVPR 2025] Official impl. of "TokenFlow: Unified Image Tokenizer for Multimodal Un···

★ 471 2026-08-09
LLaMA-Mesh 开源

用大模型直接生成 3D 网格,告别复杂建模流程

Unifying 3D Mesh Generation with Language Models

★ 1174 2026-08-09
HRM 开源

让AI学会分步思考,复杂推理不再翻车

Hierarchical Reasoning Model Official Release

★ 12667 2026-08-09
NLP_pytorch_project 开源

PyTorch NLP实战代码库,覆盖翻译、分类、生成等核心任务

Embedding, NMT, Text_Classification, Text_Generation, NER etc.

★ 565 2026-08-09
small-doge 开源

小模型也能跑得溜,低资源部署不将就。

Doge Family of Small Language Models

★ 199 2026-08-09
Time-Series-Reasoning-Survey 开源

一篇综述看懂LLM如何推理时间序列

[TMLR 2026] A Survey of Reasoning and Agentic Systems in Time Series with Large Lang···

★ 57 2026-08-09