MaskUnet

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

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 图像生成 🍴 复刻 5 👀 关注 17
访问官网 ↗ GitHub ↗
GitHub 星标 ★ 109
维护状态 低维护
是否开源 开源
定价模式 🏷️完全免费 开源项目,提供官方PyTorch实现,可自行部署使用,完全免费。

项目预览

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

项目数据

分类图像生成
开发团队gudaochangsheng
定价模式🏷️完全免费
价格说明开源项目,提供官方PyTorch实现,可自行部署使用,完全免费。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
主要语言Python
技术栈/模型diffusion-models,mask,stable-diffusion,text-to-image-generation,text-to-video-generation,u-net
GitHub 星标★ 109
上线时间2025-03-25 00:00:00
代码更新2026-05-19
仓库体积4.1 MB
本站数据更新2026-11-10
维护状态低维护
收录时间2026-08-09
浏览次数4

项目介绍

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

Star 增长趋势

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

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

核心亮点

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

不足之处

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

适用场景

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

替代项目

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

上一篇:SD-Latent-Upscaler

下一篇:comfyui-lama-remover

同类项目推荐

ComfyUI-PuLID-Flux2 开源

在ComfyUI中让FLUX.2生成同一张脸,无需训练

First PuLID implementation for FLUX.2 — Consistent face identity in ComfyUI

★ 129 2026-08-09
img2imgGAN 开源

让图像翻译不再单调,一个输入多种输出

Implementation of the paper : "Toward Multimodal Image-to-Image Translation"

★ 57 2026-08-09
ID-Patch 开源

让群体照片生成时,每个人的脸都能对号入座

Official implementation of CVPR 2025 paper "ID-Patch: Robust ID Association for Grou···

★ 77 2026-08-09
FIBO 开源

用 JSON 精确控制图像生成,安全合规,企业可放心用

FIBO is a SOTA, first open-source, JSON-native text-to-image model built for control···

★ 359 2026-08-09
color-peel 开源

把颜色从形状中剥出来,让AI按你的要求精准上色。

we propose to generate a series of geometric shapes with target colors to disentangl···

★ 68 2026-08-09
was-node-suite-comfyui 开源

给 ComfyUI 补齐上百个图像/文本/逻辑节点,工作流想怎么连就怎么连

WAS-NS v3 Reborn; Tools for image processing, filters, masking, text, logic, numbers···

★ 1895 2026-10-16
Text-to-Image-Synthesis 开源

用文本描述生成对应图像,快速跑通经典GAN基线

Pytorch implementation of Generative Adversarial Text-to-Image Synthesis paper

★ 409 2026-08-09