x-unet

用最新注意力机制升级 U-Net,提升图像生成与分割效果

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

项目预览

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

项目数据

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

项目介绍

x-unet 是一个基于 PyTorch 的 U-Net 图像生成与分割实现,旨在整合高效的注意力机制和最新的研究成果。它解决了传统 U-Net 在长距离依赖建模上的不足,通过引入高效注意力模块,在保持计算效率的同时提升模型对全局上下文的理解能力。项目提供了灵活的架构配置,支持多种注意力变体,并遵循简洁的代码风格,便于研究人员快速实验和二次开发。核心能力包括:模块化的 U-Net 组件、多种注意力机制集成、以及面向图像生成和分割任务的训练与推理接口。作为早期项目,它适合对最新注意力机制感兴趣的研究者,用于快速验证想法或作为基线模型。

Star 增长趋势

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

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

核心亮点

  • 集成了高效注意力机制,能捕获长距离依赖,优于传统 U-Net
  • 模块化设计,各组件可自由组合,便于定制和实验
  • 代码简洁,基于 PyTorch,易于上手和二次开发

不足之处

  • 文档/社区待观察
  • 项目处于早期,功能可能不完善,需自行测试

适用场景

  • 图像分割任务,如医学影像分割
  • 图像生成任务,如无条件或条件生成
  • 作为研究基线,验证新注意力机制的效果

替代项目

Segmentation Models PyTorch、MONAI、nnU-Net

同类项目推荐

FrozenBiLM 开源

冻结语言模型,零样本看懂视频回答问题

[NeurIPS 2022] Zero-Shot Video Question Answering via Frozen Bidirectional Language ···

★ 159 2026-08-09
ImageIndexer 开源

本地跑大模型,自动给图片打标签,搜图快人一步

Creates an index of images, queries a local LLM and adds tags to the image metadata

★ 399 2026-08-09
Open3D-ML 开源

开箱即用的3D点云模型训练与部署工具

An extension of Open3D to address 3D Machine Learning tasks

★ 2362 2026-08-10
ANTs 开源

医学图像精准配准与分割,助力神经影像分析

Advanced Normalization Tools (ANTs)

★ 1512 2026-08-10
visionworkbench 开源

NASA 出品的 C++ 视觉工具箱,做高精度图像配准与三维重建

The NASA Vision Workbench is a general purpose image processing and computer vision ···

★ 504 2026-09-18
PartGLEE 开源

识别并解析任意物体的部件,让AI看懂物体内部结构

[ECCV2024] PartGLEE: A Foundation Model for Recognizing and Parsing Any Objects

★ 66 2026-08-09
Falcon-Perception 开源

原生多模态模型推理,看图识文一步到位

Inference repo for Falcon-Perception and Falcon-OCR model, early-fusion, natively mu···

★ 796 2026-09-19
TeleOCR 开源

Telegram 图片自动转文字,聊天记录秒变可搜索文本

★ 368 2026-09-19