ClickAttention: Click Region Similarity Guided Interactive Segmentation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Xu, Long, Li, Shanghong, Chen, Yongquan, Chen, Junkang, Huang, Rui, Wu, Feng
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911985828888576
author Xu, Long
Li, Shanghong
Chen, Yongquan
Chen, Junkang
Huang, Rui
Wu, Feng
author_facet Xu, Long
Li, Shanghong
Chen, Yongquan
Chen, Junkang
Huang, Rui
Wu, Feng
contents Interactive segmentation algorithms based on click points have garnered significant attention from researchers in recent years. However, existing studies typically use sparse click maps as model inputs to segment specific target objects, which primarily affect local regions and have limited abilities to focus on the whole target object, leading to increased times of clicks. In addition, most existing algorithms can not balance well between high performance and efficiency. To address this issue, we propose a click attention algorithm that expands the influence range of positive clicks based on the similarity between positively-clicked regions and the whole input. We also propose a discriminative affinity loss to reduce the attention coupling between positive and negative click regions to avoid an accuracy decrease caused by mutual interference between positive and negative clicks. Extensive experiments demonstrate that our approach is superior to existing methods and achieves cutting-edge performance in fewer parameters. An interactive demo and all reproducible codes will be released at https://github.com/hahamyt/ClickAttention.
format Preprint
id arxiv_https___arxiv_org_abs_2408_06021
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle ClickAttention: Click Region Similarity Guided Interactive Segmentation
Xu, Long
Li, Shanghong
Chen, Yongquan
Chen, Junkang
Huang, Rui
Wu, Feng
Computer Vision and Pattern Recognition
Interactive segmentation algorithms based on click points have garnered significant attention from researchers in recent years. However, existing studies typically use sparse click maps as model inputs to segment specific target objects, which primarily affect local regions and have limited abilities to focus on the whole target object, leading to increased times of clicks. In addition, most existing algorithms can not balance well between high performance and efficiency. To address this issue, we propose a click attention algorithm that expands the influence range of positive clicks based on the similarity between positively-clicked regions and the whole input. We also propose a discriminative affinity loss to reduce the attention coupling between positive and negative click regions to avoid an accuracy decrease caused by mutual interference between positive and negative clicks. Extensive experiments demonstrate that our approach is superior to existing methods and achieves cutting-edge performance in fewer parameters. An interactive demo and all reproducible codes will be released at https://github.com/hahamyt/ClickAttention.
title ClickAttention: Click Region Similarity Guided Interactive Segmentation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2408.06021