MaskUnet

掩码扩散模型参数,提升图像视频生成质量

MaskUnet 是 CVPR 2025 的官方 PyTorch 实现,针对扩散模型提出一种参数掩码策略。核心思想是并非所有参数对生成同等重要,通过训练中动态掩码部分 U-Net 参数,提升模型生成能力。它适用于文本到图像和文本到视频生成任务,在保持模型结构不变的情况下,通过稀疏化训练增强特征表达和生成质量。项目提供完整训练与推理代码,基于 Stable Diffusion 架构,支持自定义掩码比例和训练配置。解决扩散模型训练效率与生成质量平衡问题,为生成模型提供一种轻量级优化手段。

开源 free 图像生成
访问官网 ↗ GitHub ↗ 文档 ↗
GitHub 星标 ★ 158
维护状态 低维护
是否开源 是
定价模式 free

项目数据

分类图像生成
开发团队gudaochangsheng
所属国家
定价模式free
价格说明开源项目,提供官方PyTorch实现,可自行部署使用,完全免费。
访问状态
是否开源是
开源协议
主要语言Python
技术栈/模型diffusion-models,mask,stable-diffusion,text-to-image-generation,text-to-video-generation,u-net
GitHub 星标★ 158
30天Star增速
HF 下载量
上线时间2025-03-25 00:00:00
最近更新2026-09-18 00:00:00
维护状态低维护
中文支持
访问方式
移动端支持
综合评分
收录时间2026-08-09
浏览次数3

使用教程

核心亮点

  • CVPR 2025 官方实现,方法有学术背书
  • 直接作用于 U-Net 参数,训练时无需额外模块
  • 同时支持文本到图像和文本到视频,适用面广

不足之处

  • 文档/社区待观察
  • 仅提供基础训练代码,缺少预训练模型和详细调参指南

适用场景

  • 学术研究:探索扩散模型参数效率与生成质量关系
  • 图像生成:在 Stable Diffusion 基础上提升文本到图像质量
  • 视频生成:用于文本到视频模型的训练优化

替代项目

Stable Diffusion、Diffusion Transformer (DiT)、U-ViT

项目介绍

MaskUnet 是图像领域的开源项目,由 gudaochangsheng 开发,2025 年首次发布。

它收录于图像分类,该分类目前共有 839 个项目,GitHub 星标数为 158。

项目已超过三个月没有代码更新,维护节奏明显放缓,最近一次代码更新于 2026-09-18。提供官方PyTorch实现,可自行部署使用,完全免费。

它主要面向的使用场景是:学术研究:探索扩散模型参数效率与生成质量关系。同类可对比的替代方案包括 Stable Diffusion、Diffusion Transformer (DiT)、U-ViT。

上一篇:SD-Latent-Upscaler

下一篇:comfyui-lama-remover

同类项目推荐

satori 开源

网页代码一键变高清图,放大不糊,不用截图

Enlightened library to convert HTML and CSS to SVG

★ 13967 2026-08-09
C2GAN 开源

用关键点精准控制图像生成,让AI画出你想要的结构

[ACM MM 2019 Oral] Cycle In Cycle Generative Adversarial Networks for Keypoint-Guide···

★ 70 2026-08-09
ComfyUI 开源

像搭积木一样连节点,生图流程自己搭

The most powerful and modular diffusion model GUI, api and backend with a graph/node···

★ 134493 2026-08-09