retrieval-augmented-ddpm

给扩散模型加个检索外挂,生成更准更丰富

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

项目预览

retrieval-augmented-ddpm 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

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

项目介绍

这是一个基于 PyTorch 实现的检索增强去噪扩散概率模型(Retrieval-Augmented DDPM)。它解决的是传统扩散模型在生成图像时依赖单一训练分布、缺乏外部知识引导的问题。核心能力是在 DDPM 的采样或训练过程中引入检索机制,从外部数据库或训练集中检索相似样本作为条件,从而提升生成图像的多样性和保真度。项目提供了完整的模型定义、训练与采样流程,适合研究者复现和扩展。

Star 增长趋势

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

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

核心亮点

  • 将检索机制与扩散模型结合,为生成过程提供外部参考,思路新颖
  • 基于 PyTorch 实现,代码结构清晰,便于二次开发
  • 支持自定义检索库,灵活适配不同数据场景

不足之处

  • 项目处于早期阶段,文档和示例较少
  • 检索模块依赖外部数据库,可能增加推理开销

适用场景

  • 学术研究:探索检索增强生成模型的新方法
  • 图像生成:需要从特定参考集生成相似风格或内容的图像
  • 数据增强:利用检索库提升生成样本的多样性

替代项目

Retrieval-Augmented Diffusion Models (RDM)、KNN-Diffusion、Re-Imagen

上一篇:gansformer

下一篇:RQ-Transformer

同类项目推荐

big-sleep 开源

输入一句话,用 CLIP 引导 BigGAN 生成图像

A simple command line tool for text to image generation, using OpenAI's CLIP and a B···

★ 2574 2026-08-09
stable-diffusion-discord-bot 开源

在 Discord 里发条消息,就能用 Stable Diffusion 生成 AI 画作

A discord bot to generate AI art from prompts using invokeAI

★ 211 2026-08-09
StabilityMatrix 开源

一站式管理 Stable Diffusion 环境与模型,告别命令行折腾

Multi-Platform Package Manager for Stable Diffusion

★ 9002 2026-08-09
Forgedit 开源

用一句话改图,不用蒙版,AI 帮你搞定

The first paper discovered and applied Diffusion UNet's Disentanglement on structure···

★ 284 2026-08-09
sd-webui-supermerger 开源

把多个大模型揉成一个,画风角色自由混搭

model merge extention for stable diffusion web ui

★ 837 2026-09-18
booru2prompt 开源

一键把 booru 图片帖子变成 SD 提示词,省去手抄标签

An extension for stable-diffusion-webui to convert image booru posts into prompts

★ 61 2026-08-09
PhotoShow 开源

鼠标悬停缩略图,直接看高清大图并一键下载

PhotoShow is a browser extension helping you view/copy/download high-definition imag···

★ 442 2026-10-22