A Simple Image Segmentation Framework via In-Context Examples

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Liu, Yang, Jing, Chenchen, Li, Hengtao, Zhu, Muzhi, Chen, Hao, Wang, Xinlong, Shen, Chunhua
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866917797325438976
author Liu, Yang
Jing, Chenchen
Li, Hengtao
Zhu, Muzhi
Chen, Hao
Wang, Xinlong
Shen, Chunhua
author_facet Liu, Yang
Jing, Chenchen
Li, Hengtao
Zhu, Muzhi
Chen, Hao
Wang, Xinlong
Shen, Chunhua
contents Recently, there have been explorations of generalist segmentation models that can effectively tackle a variety of image segmentation tasks within a unified in-context learning framework. However, these methods still struggle with task ambiguity in in-context segmentation, as not all in-context examples can accurately convey the task information. In order to address this issue, we present SINE, a simple image Segmentation framework utilizing in-context examples. Our approach leverages a Transformer encoder-decoder structure, where the encoder provides high-quality image representations, and the decoder is designed to yield multiple task-specific output masks to effectively eliminate task ambiguity. Specifically, we introduce an In-context Interaction module to complement in-context information and produce correlations between the target image and the in-context example and a Matching Transformer that uses fixed matching and a Hungarian algorithm to eliminate differences between different tasks. In addition, we have further perfected the current evaluation system for in-context image segmentation, aiming to facilitate a holistic appraisal of these models. Experiments on various segmentation tasks show the effectiveness of the proposed method.
format Preprint
id arxiv_https___arxiv_org_abs_2410_04842
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle A Simple Image Segmentation Framework via In-Context Examples
Liu, Yang
Jing, Chenchen
Li, Hengtao
Zhu, Muzhi
Chen, Hao
Wang, Xinlong
Shen, Chunhua
Computer Vision and Pattern Recognition
Recently, there have been explorations of generalist segmentation models that can effectively tackle a variety of image segmentation tasks within a unified in-context learning framework. However, these methods still struggle with task ambiguity in in-context segmentation, as not all in-context examples can accurately convey the task information. In order to address this issue, we present SINE, a simple image Segmentation framework utilizing in-context examples. Our approach leverages a Transformer encoder-decoder structure, where the encoder provides high-quality image representations, and the decoder is designed to yield multiple task-specific output masks to effectively eliminate task ambiguity. Specifically, we introduce an In-context Interaction module to complement in-context information and produce correlations between the target image and the in-context example and a Matching Transformer that uses fixed matching and a Hungarian algorithm to eliminate differences between different tasks. In addition, we have further perfected the current evaluation system for in-context image segmentation, aiming to facilitate a holistic appraisal of these models. Experiments on various segmentation tasks show the effectiveness of the proposed method.
title A Simple Image Segmentation Framework via In-Context Examples
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2410.04842