网易首页 > 网易号 > 正文 申请入驻

统一图像生成,无需繁杂插件!智源发布扩散模型框架OmniGe

0
分享至

多模态模型,统一图像生成。

最新扩散模型框架来了。

智源研究院推出OmniGen,它能天然支持各种图像生成任务,架构高度简化,还能有效跨不同任务迁移知识,应对未见过的任务和领域。

特点如下:

1、统一性:OmniGen 天然地支持各种图像生成任务,例如文生图、图像编辑、主题驱动生成和视觉条件生成等。此外,OmniGen可以处理经典的计算机视觉任务,将其转换为图像生成任务。

2、简单性:OmniGen 的架构高度简化。此外,与现有模型相比,它更加用户友好,可以通过指令完成复杂的任务,而不需要冗长的处理步骤和额外的模块(如 ControlNet 或 IP-Adapter),从而大大简化了工作流程。

3、知识迁移:受益于统一格式的学习,OmniGen 有效地跨不同任务迁移知识,应对未见过的任务和领域,并展示新颖的功能。我们还探讨了模型的推理能力和思维链机制的在图像生成领域的潜在应用。

基于 OmniGen 的通用能力,可实施更灵活的图像生成,以下展示一个简单 Pipeline:文本生成图像,编辑生成图像的部分元素,根据生成图像的人体姿态生成重绘图像,从另一图像中提取所需对象与新图像融合。

AI生图插件操作繁琐

近年来,许多文生图模型在生成式 AI 的浪潮中脱颖而出。然而,这些出色的专有模型仅能基于文本生成图像。当用户产生更灵活、复杂、精细等的图像生成需求时,往往需要额外的插件和操作。

例如,若想参考任一姿态生成图像,常规方法是:用姿态检测器从参考图像中估计姿态作为条件输入,并加载对应的 Controlnet 插件,最后提取条件输入的特征馈入扩散模型生成图像。

此外,若想基于合照中的特定人物生成新图像,流程更加繁琐,需要裁剪图像以确保结果图像仅包含目标人物。

而诸如 InstandID 等方法还需使用额外的人脸检测器提取面部信息,并用面部编码器提取特征以输入模型。

值得注意的是,各种不同的生成任务甚至还需更多不同的插件和操作,如此复杂、琐碎而冗长的工作流极大地增加了训练和应用的成本。然而,即便如此繁琐,有时也仍难以满足一般的图像生成的需求,例如基于指定多张照片中的实体生成新图像。

相反,在文本生成领域,以 ChatGPT 为代表的模型可通过人类指令直接处理各种文本任务。那么,在图像生成领域,能否通过单个支持多种输入且耦合多项能力的模型,基于用户指令完成各种生成任务,而无需各种繁杂的流程吗?

为解决这一挑战性问题,智源发布了统一图像生成模型 OmniGen。OmniGen 模型具有良好的简洁性和易用性,集成了多种基础图像生成任务,包括但不限于:文生图、图像编辑、角色一致性生成、基于视觉条件的生成等。

OmniGen 支持基于任意多模态的文图指令完成任务,而无需任何其他额外插件和操作。

强大通用能力

OmniGen 集多项能力于一体,包括但不限于:

以下简要展示部分能力效果:

2.1 文本到图像生成

2.2 指代表达生成

OmniGen 具备类似 InstandID、Pulid 等模型生成角色一致性图像等能力,即:输入具有单个对象的图像,理解并遵循指令,输出基于该对象的新图像。同时,OmniGen 具有更高阶的能力:指代表达生成能力,我们把这种能力定义为能够从包含多个对象的图像中,识别指令所指代的对象并生成新的图像。例如,OmniGen 可根据指令直接从多人图像中定位目标对象,并生成遵循指令的新图像,而无需任何额外的模块和操作:

更多样例:

2.3 通用图像条件生成

OmniGen 不仅支持类似 ControlNet 根据特定显式条件生成图像的能力,还同时具备处理经典计算机视觉任务的能力(如人体姿态估计、深度估计等)。

因此,OmniGen 可凭借单个模型完成整个 ControlNet 流程:直接使用 OmniGen 对原图提取视觉条件,并基于所提取的条件生成图像,无需额外处理器。

同时,OmniGen 还能更进一步简化中间流程,一步出图:直接输入原图,输入指令“Following the human pose(or depth mapping) of this image, generate a new image:…”,就可根据输入图像的人体姿态或深度图关系生成新图像。

2.4 图像编辑

OmniGen 具备良好的图像编辑能力,并且可以在一次运行中同时执行多条编辑指令,例如:

2.5 更多能力

OmniGen 具备潜在的推理能力,可以处理对模型理解和推断能力具有一定要求的非显式查询指令。

例如,要求模型删除图中能装水的物品,则模型能够理解和推断出指令涉及的图中物体并删除:

另一方面,OmniGen 具有一定程度的上下文学习能力,可根据参考样例对图像进行处理。例如,输入一个分割皇后象棋的输入-输出配对样例 (Example),模型能识别并分割新输入图像中对应的物体:

思维链(Chain-of-Thought, CoT)方法将任务分解为多个步骤,并按顺序求解每个步骤以获得准确的最终答案,从而显著提高了llm的性能。我们考虑是否可以将类似的替代方案应用于图像生成。受人类绘画的基本方式的启发,我们希望模仿一步一步的绘画过程,从空白画布上迭代地生成图像。我们进行了初步的探索,微调后模型能够模拟人类行为一步步的生成图片,进一步的优化留给以后的研究。

OmniGen 的能力包括但不限于以上内容,还包括基本的图像去噪、边缘提取等能力。模型权重和代码已开源,用户可以自行探索更多OmniGen的能力。

文章转载:[xo.growlightkitz.com)

最大程度舍弃额外模块

文章转载:[ty.gracesmiths.com)

OmniGen 的核心设计原则是:简洁和有效。

文章转载:[pim.genf20shop.com)

因此,研究团队最大程度舍弃了各种额外模块。OmniGen 的基本架构为:一个 Transformer 模型和一个 VAE 模块,共 3.8B 参数。其中,Transformer 继承于 Phi3-mini 模型,图像内部改用双向注意力 (Bidirectional Attention) 以契合图像数据特性。整体架构如下所示:

文章转载:[ls.geneloveland.com)

为实现强大的通用和泛化能力,研究人员需要基于大规模和多样化的数据集训练模型。然而,在图像生成领域,尚无一个可用的通用数据集。为此,我们构建了首个大规模且多样化的统一图像生成数据集 X2I,意为 “Anything to Image”。其中,不同任务的数据格式被重新组织和统一,以便于管理和使用。X2I 数据集包含约 1 亿图像,未来经审查等流程后将开源,旨在进一步推动通用图像生成领域的发展。下图简要展示了 X2I 数据集的一些示例:

文章转载:[hcox.gamerbutton.com)

总之,OmniGen的统一图像生成范式,不但有助于执行各种下游任务,而且有利于组合各种能力满足更通用的需求。当前,OmniGen 的报告、权重和代码等已开源,欢迎社区共同参与对 OmniGen 潜在能力的发掘、基本性能的提升和广泛应用的探索。

文章转载:[dmg.fx8000.com)

OmniGen 模型是对统一图像生成的初步尝试,还有很大的提升空间。未来,智源将进一步改进模型基本能力,拓展更多有趣的功能。同时,微调代码已发布,用户可简单对其进行微调,由于OmniGen的输入形式非常多样,用户可自行定义各式各样的微调任务,赋予模型更多有意思的能力。

文章转载:[zwy.es14.com)

Paper: https://arxiv.org/pdf/2409.11340

文章转载:[vgie.empeach.com)

Code: https://github.com/VectorSpaceLab/OmniGen

文章转载:[rqay.donkingmedia.com)

Demo: https://huggingface.co/spaces/Shitao/OmniGen

文章转载:[nals.cyhere.com)

特别声明:以上内容(如有图片或视频亦包括在内)为自媒体平台“网易号”用户上传并发布,本平台仅提供信息存储服务。

Notice: The content above (including the pictures and videos if any) is uploaded and posted by a user of NetEase Hao, which is a social media platform and only provides information storage services.

相关推荐
热点推荐
范冰冰饭局照流出,喝酒还吐舌头,素颜温婉大方,和妆后判若两人

范冰冰饭局照流出,喝酒还吐舌头,素颜温婉大方,和妆后判若两人

南城无双
2024-10-25 23:54:18
福克斯谈60分:若常规时间赢球拿51分就行 我一直在攻筐&太累了

福克斯谈60分:若常规时间赢球拿51分就行 我一直在攻筐&太累了

直播吧
2024-12-21 23:50:22
河南:女子常年打扮辣眼,不管走哪都是亮点,家人:没人管得了她

河南:女子常年打扮辣眼,不管走哪都是亮点,家人:没人管得了她

丹宝说文史
2023-11-10 15:41:32
上海迪士尼:近2/3到访游客游玩了“疯狂动物城:热力追踪”项目

上海迪士尼:近2/3到访游客游玩了“疯狂动物城:热力追踪”项目

北京商报
2024-12-20 13:06:20
如果孩子厌学,内驱力不足,家长一定要带孩子看这两部高分纪录片!

如果孩子厌学,内驱力不足,家长一定要带孩子看这两部高分纪录片!

阿呆爸
2024-12-21 19:05:15
曝吴奇隆刘诗诗已离婚!女方不愿再贴补男方,年底工作多将进新组

曝吴奇隆刘诗诗已离婚!女方不愿再贴补男方,年底工作多将进新组

萌神木木
2024-12-18 12:41:57
失控了!俄乌战争,最危险时刻到了!

失控了!俄乌战争,最危险时刻到了!

大嘴说天下
2024-12-21 22:19:37
毛泽东水准最高的10首诗词,乃世界之经典,一生必读!

毛泽东水准最高的10首诗词,乃世界之经典,一生必读!

诗词中国
2024-09-13 10:52:25
郑爽国外为儿子庆生,素颜出镜发量稀疏变素人,母爱满满自己带娃

郑爽国外为儿子庆生,素颜出镜发量稀疏变素人,母爱满满自己带娃

小咪侃娱圈
2024-12-20 09:30:09
亲密出游,陈幸同恋情曝光?男方27岁,型男,和樊振东发小

亲密出游,陈幸同恋情曝光?男方27岁,型男,和樊振东发小

东球弟
2024-12-22 10:44:47
里程碑!哈登总分超邓肯升历史第16 现役第3仅输詹姆斯杜兰特

里程碑!哈登总分超邓肯升历史第16 现役第3仅输詹姆斯杜兰特

醉卧浮生
2024-12-22 10:34:55
45岁高圆圆穿上紧身针织衫,才发现,原来她的身材竟然如此之好

45岁高圆圆穿上紧身针织衫,才发现,原来她的身材竟然如此之好

南城无双
2024-12-01 00:26:09
小米SU7坠崖,雷军又一次封神了,我要笑死在网友评论区里

小米SU7坠崖,雷军又一次封神了,我要笑死在网友评论区里

沙雕小琳琳
2024-12-22 12:09:09
原产于中国的螳螂,正在美国进行平等的杀戮

原产于中国的螳螂,正在美国进行平等的杀戮

果壳
2024-12-20 12:13:14
为啥银行不给办一类卡?网友分享太真实了!解释的通俗易懂

为啥银行不给办一类卡?网友分享太真实了!解释的通俗易懂

有趣的火烈鸟
2024-12-21 17:12:13
震惊!浙大学生拒签后怒言:这辈子不会放过!学校已报警

震惊!浙大学生拒签后怒言:这辈子不会放过!学校已报警

魔都姐姐杂谈
2024-12-19 11:38:13
临川一中原校长王昱去世,年仅57岁,亲戚公开死因,学生曝其人品

临川一中原校长王昱去世,年仅57岁,亲戚公开死因,学生曝其人品

180°视角
2024-12-19 15:19:40
外媒:中国近期购买125万吨美国大豆

外媒:中国近期购买125万吨美国大豆

参考消息
2024-12-22 11:51:11
家有儿子的要知道:家里什么方位旺儿子,小时候有官相的特征

家有儿子的要知道:家里什么方位旺儿子,小时候有官相的特征

千年人参它会跑
2024-07-13 08:19:47
为什么我感受不到1500元的手机比四五千的差?看网友评论终于懂了

为什么我感受不到1500元的手机比四五千的差?看网友评论终于懂了

奇特短尾矮袋鼠
2024-08-02 07:22:23
2024-12-22 15:56:49

科技要闻

售价78.80万元起,蔚来高端行政车ET9上市

头条要闻

80岁的全球顶级富豪又结婚了 33岁的华裔妻子身份披露

头条要闻

80岁的全球顶级富豪又结婚了 33岁的华裔妻子身份披露

体育要闻

逆潮流而行!灰熊已成NBA最另类球队

娱乐要闻

黄圣依与杨子被曝未领结婚证,净身出户真相曝光

财经要闻

山姆,又出事!网友:太吓人

汽车要闻

直面78S挑战 蔚来ET9售价66万元起

态度原创

本地
时尚
房产
旅游
军事航空

本地新闻

好吃潮州|潮州腐乳饼,咸甜党都沦陷了

厚外套里面穿什么?这么穿显瘦洋气,超好看

房产要闻

刚刚,三亚万科官宣:首开爆卖40.8亿!

旅游要闻

大片来袭!崂山巨峰再现“雾凇”奇观

军事要闻

扎哈罗娃:乌克兰无人机袭击俄罗斯喀山系报复

无障碍浏览 进入关怀版