Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

T2I-Adapter: Learning Adapters to Dig out More Controllable Ability for Text-to-Image Diffusion Models

About

The incredible generative ability of large-scale text-to-image (T2I) models has demonstrated strong power of learning complex structures and meaningful semantics. However, relying solely on text prompts cannot fully take advantage of the knowledge learned by the model, especially when flexible and accurate controlling (e.g., color and structure) is needed. In this paper, we aim to ``dig out" the capabilities that T2I models have implicitly learned, and then explicitly use them to control the generation more granularly. Specifically, we propose to learn simple and lightweight T2I-Adapters to align internal knowledge in T2I models with external control signals, while freezing the original large T2I models. In this way, we can train various adapters according to different conditions, achieving rich control and editing effects in the color and structure of the generation results. Further, the proposed T2I-Adapters have attractive properties of practical value, such as composability and generalization ability. Extensive experiments demonstrate that our T2I-Adapter has promising generation quality and a wide range of applications.

Chong Mou, Xintao Wang, Liangbin Xie, Yanze Wu, Jian Zhang, Zhongang Qi, Ying Shan, Xiaohu Qie• 2023

Related benchmarks

TaskDatasetResultRank
Polyp SegmentationCVC-ClinicDB (test)
DSC93.7
196
Polyp SegmentationKvasir
Dice Score91.1
128
Polyp SegmentationETIS
Dice Score78.1
108
Polyp SegmentationETIS (test)
Mean Dice81.1
86
Skin Lesion SegmentationISIC 2018 (test)
Dice Score91.19
74
Polyp SegmentationColonDB
mDice79.6
74
Polyp SegmentationKvasir (test)
Dice Coefficient91.9
73
Polyp SegmentationEndoScene
mDice89.2
61
SegmentationADE20K
mIoU16.38
52
Polyp SegmentationClinicDB
mDice0.94
50
Showing 10 of 101 rows
...

Other info

Follow for update