ShapeFormer: Shape Prior Visible-to-Amodal Transformer-based Amodal Instance Segmentation

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Tran, Minh, Bounsavy, Winston, Vo, Khoa, Nguyen, Anh, Nguyen, Tri, Le, Ngan
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866914758025805824
author Tran, Minh
Bounsavy, Winston
Vo, Khoa
Nguyen, Anh
Nguyen, Tri
Le, Ngan
author_facet Tran, Minh
Bounsavy, Winston
Vo, Khoa
Nguyen, Anh
Nguyen, Tri
Le, Ngan
contents Amodal Instance Segmentation (AIS) presents a challenging task as it involves predicting both visible and occluded parts of objects within images. Existing AIS methods rely on a bidirectional approach, encompassing both the transition from amodal features to visible features (amodal-to-visible) and from visible features to amodal features (visible-to-amodal). Our observation shows that the utilization of amodal features through the amodal-to-visible can confuse the visible features due to the extra information of occluded/hidden segments not presented in visible display. Consequently, this compromised quality of visible features during the subsequent visible-to-amodal transition. To tackle this issue, we introduce ShapeFormer, a decoupled Transformer-based model with a visible-to-amodal transition. It facilitates the explicit relationship between output segmentations and avoids the need for amodal-to-visible transitions. ShapeFormer comprises three key modules: (i) Visible-Occluding Mask Head for predicting visible segmentation with occlusion awareness, (ii) Shape-Prior Amodal Mask Head for predicting amodal and occluded masks, and (iii) Category-Specific Shape Prior Retriever aims to provide shape prior knowledge. Comprehensive experiments and extensive ablation studies across various AIS benchmarks demonstrate the effectiveness of our ShapeFormer. The code is available at: \url{https://github.com/UARK-AICV/ShapeFormer}
format Preprint
id arxiv_https___arxiv_org_abs_2403_11376
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle ShapeFormer: Shape Prior Visible-to-Amodal Transformer-based Amodal Instance Segmentation
Tran, Minh
Bounsavy, Winston
Vo, Khoa
Nguyen, Anh
Nguyen, Tri
Le, Ngan
Computer Vision and Pattern Recognition
Amodal Instance Segmentation (AIS) presents a challenging task as it involves predicting both visible and occluded parts of objects within images. Existing AIS methods rely on a bidirectional approach, encompassing both the transition from amodal features to visible features (amodal-to-visible) and from visible features to amodal features (visible-to-amodal). Our observation shows that the utilization of amodal features through the amodal-to-visible can confuse the visible features due to the extra information of occluded/hidden segments not presented in visible display. Consequently, this compromised quality of visible features during the subsequent visible-to-amodal transition. To tackle this issue, we introduce ShapeFormer, a decoupled Transformer-based model with a visible-to-amodal transition. It facilitates the explicit relationship between output segmentations and avoids the need for amodal-to-visible transitions. ShapeFormer comprises three key modules: (i) Visible-Occluding Mask Head for predicting visible segmentation with occlusion awareness, (ii) Shape-Prior Amodal Mask Head for predicting amodal and occluded masks, and (iii) Category-Specific Shape Prior Retriever aims to provide shape prior knowledge. Comprehensive experiments and extensive ablation studies across various AIS benchmarks demonstrate the effectiveness of our ShapeFormer. The code is available at: \url{https://github.com/UARK-AICV/ShapeFormer}
title ShapeFormer: Shape Prior Visible-to-Amodal Transformer-based Amodal Instance Segmentation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2403.11376