DreamRelation: Bridging Customization and Relation Generation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Shi, Qingyu, Qi, Lu, Wu, Jianzong, Bai, Jinbin, Wang, Jingbo, Tong, Yunhai, Li, Xiangtai
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866910903888248832
author Shi, Qingyu
Qi, Lu
Wu, Jianzong
Bai, Jinbin
Wang, Jingbo
Tong, Yunhai
Li, Xiangtai
author_facet Shi, Qingyu
Qi, Lu
Wu, Jianzong
Bai, Jinbin
Wang, Jingbo
Tong, Yunhai
Li, Xiangtai
contents Customized image generation is essential for creating personalized content based on user prompts, allowing large-scale text-to-image diffusion models to more effectively meet individual needs. However, existing models often neglect the relationships between customized objects in generated images. In contrast, this work addresses this gap by focusing on relation-aware customized image generation, which seeks to preserve the identities from image prompts while maintaining the relationship specified in text prompts. Specifically, we introduce DreamRelation, a framework that disentangles identity and relation learning using a carefully curated dataset. Our training data consists of relation-specific images, independent object images containing identity information, and text prompts to guide relation generation. Then, we propose two key modules to tackle the two main challenges: generating accurate and natural relationships, especially when significant pose adjustments are required, and avoiding object confusion in cases of overlap. First, we introduce a keypoint matching loss that effectively guides the model in adjusting object poses closely tied to their relationships. Second, we incorporate local features of the image prompts to better distinguish between objects, preventing confusion in overlapping cases. Extensive results on our proposed benchmarks demonstrate the superiority of DreamRelation in generating precise relations while preserving object identities across a diverse set of objects and relationships.
format Preprint
id arxiv_https___arxiv_org_abs_2410_23280
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle DreamRelation: Bridging Customization and Relation Generation
Shi, Qingyu
Qi, Lu
Wu, Jianzong
Bai, Jinbin
Wang, Jingbo
Tong, Yunhai
Li, Xiangtai
Computer Vision and Pattern Recognition
Customized image generation is essential for creating personalized content based on user prompts, allowing large-scale text-to-image diffusion models to more effectively meet individual needs. However, existing models often neglect the relationships between customized objects in generated images. In contrast, this work addresses this gap by focusing on relation-aware customized image generation, which seeks to preserve the identities from image prompts while maintaining the relationship specified in text prompts. Specifically, we introduce DreamRelation, a framework that disentangles identity and relation learning using a carefully curated dataset. Our training data consists of relation-specific images, independent object images containing identity information, and text prompts to guide relation generation. Then, we propose two key modules to tackle the two main challenges: generating accurate and natural relationships, especially when significant pose adjustments are required, and avoiding object confusion in cases of overlap. First, we introduce a keypoint matching loss that effectively guides the model in adjusting object poses closely tied to their relationships. Second, we incorporate local features of the image prompts to better distinguish between objects, preventing confusion in overlapping cases. Extensive results on our proposed benchmarks demonstrate the superiority of DreamRelation in generating precise relations while preserving object identities across a diverse set of objects and relationships.
title DreamRelation: Bridging Customization and Relation Generation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2410.23280