CFR-ICL: Cascade-Forward Refinement with Iterative Click Loss for Interactive Image Segmentation

9 Mar 2023  ยท  Shoukun Sun, Min Xian, Fei Xu, Luca Capriotti, Tiankai Yao ยท

The click-based interactive segmentation aims to extract the object of interest from an image with the guidance of user clicks. Recent work has achieved great overall performance by employing feedback from the output. However, in most state-of-the-art approaches, 1) the inference stage involves inflexible heuristic rules and requires a separate refinement model, and 2) the number of user clicks and model performance cannot be balanced. To address the challenges, we propose a click-based and mask-guided interactive image segmentation framework containing three novel components: Cascade-Forward Refinement (CFR), Iterative Click Loss (ICL), and SUEM image augmentation. The CFR offers a unified inference framework to generate segmentation results in a coarse-to-fine manner. The proposed ICL allows model training to improve segmentation and reduce user interactions simultaneously. The proposed SUEM augmentation is a comprehensive way to create large and diverse training sets for interactive image segmentation. Extensive experiments demonstrate the state-of-the-art performance of the proposed approach on five public datasets. Remarkably, our model reduces by 33.2\%, and 15.5\% the number of clicks required to surpass an IoU of 0.95 in the previous state-of-the-art approach on the Berkeley and DAVIS sets, respectively.

PDF Abstract

Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Interactive Segmentation Berkeley ICL CFR-1 (ViT-H, C+L) NoC@90 1.46 # 1
NoC@95 2.90 # 1
Interactive Segmentation DAVIS ICL CFR-1 (ViT-H, C+L) NoC@85 3 # 2
NoC@90 4.24 # 1
NoC@95 7.50 # 1
Interactive Segmentation GrabCut SimpleClick CFR-1 (ViT-H, SBD) NoC@85 1.30 # 1
NoC@90 1.32 # 1
NoC@95 1.78 # 2
Interactive Segmentation GrabCut ICL CFR-1 (ViT-H, SBD) NoC@90 1.42 # 3
NoC@95 1.62 # 1
Interactive Segmentation PASCAL VOC ICL CFR-1 (ViT-H, C+L) NoC@85 1.72 # 1
NoC@90 1.94 # 1
NoC@95 2.45 # 1
Interactive Segmentation SBD SimpleClick CFR-1 (ViT-H, SBD) NoC@85 2.45 # 1
NoC@90 4.08 # 1
NoC@95 9.80 # 1

Methods


No methods listed for this paper. Add relevant methods here