VRP-SAM: SAM with Visual Reference Prompt

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Sun, Yanpeng, Chen, Jiahui, Zhang, Shan, Zhang, Xinyu, Li, Xiaofan, Chen, Qiang, Zhang, Gang, Ding, Errui, Wang, Jingdong, Li, Zechao
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866914128390520832
author Sun, Yanpeng
Chen, Jiahui
Zhang, Shan
Zhang, Xinyu
Li, Xiaofan
Chen, Qiang
Zhang, Gang
Ding, Errui
Wang, Jingdong
Li, Zechao
author_facet Sun, Yanpeng
Chen, Jiahui
Zhang, Shan
Zhang, Xinyu
Li, Xiaofan
Chen, Qiang
Zhang, Gang
Ding, Errui
Wang, Jingdong
Li, Zechao
contents In this paper, we propose a novel Visual Reference Prompt (VRP) encoder that empowers the Segment Anything Model (SAM) to utilize annotated reference images as prompts for segmentation, creating the VRP-SAM model. In essence, VRP-SAM can utilize annotated reference images to comprehend specific objects and perform segmentation of specific objects in target image. It is note that the VRP encoder can support a variety of annotation formats for reference images, including \textbf{point}, \textbf{box}, \textbf{scribble}, and \textbf{mask}. VRP-SAM achieves a breakthrough within the SAM framework by extending its versatility and applicability while preserving SAM's inherent strengths, thus enhancing user-friendliness. To enhance the generalization ability of VRP-SAM, the VRP encoder adopts a meta-learning strategy. To validate the effectiveness of VRP-SAM, we conducted extensive empirical studies on the Pascal and COCO datasets. Remarkably, VRP-SAM achieved state-of-the-art performance in visual reference segmentation with minimal learnable parameters. Furthermore, VRP-SAM demonstrates strong generalization capabilities, allowing it to perform segmentation of unseen objects and enabling cross-domain segmentation. The source code and models will be available at https://github.com/syp2ysy/VRP-SAM
format Preprint
id arxiv_https___arxiv_org_abs_2402_17726
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle VRP-SAM: SAM with Visual Reference Prompt
Sun, Yanpeng
Chen, Jiahui
Zhang, Shan
Zhang, Xinyu
Li, Xiaofan
Chen, Qiang
Zhang, Gang
Ding, Errui
Wang, Jingdong
Li, Zechao
Computer Vision and Pattern Recognition
In this paper, we propose a novel Visual Reference Prompt (VRP) encoder that empowers the Segment Anything Model (SAM) to utilize annotated reference images as prompts for segmentation, creating the VRP-SAM model. In essence, VRP-SAM can utilize annotated reference images to comprehend specific objects and perform segmentation of specific objects in target image. It is note that the VRP encoder can support a variety of annotation formats for reference images, including \textbf{point}, \textbf{box}, \textbf{scribble}, and \textbf{mask}. VRP-SAM achieves a breakthrough within the SAM framework by extending its versatility and applicability while preserving SAM's inherent strengths, thus enhancing user-friendliness. To enhance the generalization ability of VRP-SAM, the VRP encoder adopts a meta-learning strategy. To validate the effectiveness of VRP-SAM, we conducted extensive empirical studies on the Pascal and COCO datasets. Remarkably, VRP-SAM achieved state-of-the-art performance in visual reference segmentation with minimal learnable parameters. Furthermore, VRP-SAM demonstrates strong generalization capabilities, allowing it to perform segmentation of unseen objects and enabling cross-domain segmentation. The source code and models will be available at https://github.com/syp2ysy/VRP-SAM
title VRP-SAM: SAM with Visual Reference Prompt
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2402.17726