A Task is Worth One Word: Learning with Task Prompts for High-Quality Versatile Image Inpainting

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Zhuang, Junhao, Zeng, Yanhong, Liu, Wenran, Yuan, Chun, Chen, Kai
Formato: Preprint
Publicado: 2023
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866916332279169024
author Zhuang, Junhao
Zeng, Yanhong
Liu, Wenran
Yuan, Chun
Chen, Kai
author_facet Zhuang, Junhao
Zeng, Yanhong
Liu, Wenran
Yuan, Chun
Chen, Kai
contents Advancing image inpainting is challenging as it requires filling user-specified regions for various intents, such as background filling and object synthesis. Existing approaches focus on either context-aware filling or object synthesis using text descriptions. However, achieving both tasks simultaneously is challenging due to differing training strategies. To overcome this challenge, we introduce PowerPaint, the first high-quality and versatile inpainting model that excels in multiple inpainting tasks. First, we introduce learnable task prompts along with tailored fine-tuning strategies to guide the model's focus on different inpainting targets explicitly. This enables PowerPaint to accomplish various inpainting tasks by utilizing different task prompts, resulting in state-of-the-art performance. Second, we demonstrate the versatility of the task prompt in PowerPaint by showcasing its effectiveness as a negative prompt for object removal. Moreover, we leverage prompt interpolation techniques to enable controllable shape-guided object inpainting, enhancing the model's applicability in shape-guided applications. Finally, we conduct extensive experiments and applications to verify the effectiveness of PowerPaint. We release our codes and models on our project page: https://powerpaint.github.io/.
format Preprint
id arxiv_https___arxiv_org_abs_2312_03594
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle A Task is Worth One Word: Learning with Task Prompts for High-Quality Versatile Image Inpainting
Zhuang, Junhao
Zeng, Yanhong
Liu, Wenran
Yuan, Chun
Chen, Kai
Computer Vision and Pattern Recognition
Advancing image inpainting is challenging as it requires filling user-specified regions for various intents, such as background filling and object synthesis. Existing approaches focus on either context-aware filling or object synthesis using text descriptions. However, achieving both tasks simultaneously is challenging due to differing training strategies. To overcome this challenge, we introduce PowerPaint, the first high-quality and versatile inpainting model that excels in multiple inpainting tasks. First, we introduce learnable task prompts along with tailored fine-tuning strategies to guide the model's focus on different inpainting targets explicitly. This enables PowerPaint to accomplish various inpainting tasks by utilizing different task prompts, resulting in state-of-the-art performance. Second, we demonstrate the versatility of the task prompt in PowerPaint by showcasing its effectiveness as a negative prompt for object removal. Moreover, we leverage prompt interpolation techniques to enable controllable shape-guided object inpainting, enhancing the model's applicability in shape-guided applications. Finally, we conduct extensive experiments and applications to verify the effectiveness of PowerPaint. We release our codes and models on our project page: https://powerpaint.github.io/.
title A Task is Worth One Word: Learning with Task Prompts for High-Quality Versatile Image Inpainting
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2312.03594