arrow
Return

UniInst: Unique representation for end-to-end instance segmentation

delete2022-12-01
delete3
delete
OA
AI
Y
Yimin Ou
R
Rui Yang
L
Lufan Ma
Y
Yong Liu *
J
Jiangpeng Yan
C
Chengjie Wang
李秀 (Xiu Li) *
DOI:10.1016/j.neucom.2022.09.112delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
Existing instance segmentation methods have achieved impressive performance but still suffer from a common dilemma: redundant representations (e.g., multiple boxes, grids, and anchor points) are inferred for one instance, which leads to multiple duplicated predictions. Thus, mainstream methods usually rely on a hand-designed non-maximum suppression (NMS) post-processing step to select the optimal predic-tion result, consequently hindering end-to-end training. To address this issue, we propose a box-free and NMS-free end-to-end instance segmentation framework, dubbed UniInst, which yields only one unique representation for each instance. Specifically, we design an instance-aware one-to-one assignment scheme, named Only Yield One Representation (OYOR). It dynamically assigns one unique representation to each instance according to the matching quality between predictions and ground truths. Then, a novel prediction re-ranking strategy is elegantly integrated into the framework to address the misalignment between the classification score and mask quality, enabling the learned representation to be more dis-criminative. With these techniques, our UniInst, the first FCN-based box-free and NMS-free end-to-end instance segmentation framework, achieves competitive performance, e.g., 39.0 mask AP using ResNet-50-FPN and 40.2 mask AP using ResNet-101-FPN on COCO test-dev. Moreover, the proposed instance-aware method is robust to occlusion scenes because of non-dependent on box and NMS. It out-performs common baselines by remarkable mask AP on the heavily-occluded OCHuman benchmark. Code is available at https://github.com/b03505036/UniInst.(c) 2022 Published by Elsevier B.V.
Keywords:
Instance segmentation
End-to-end instance segmentation
Fully convolutional networks
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

Neurocomputing cover
Neurocomputing
IF:
6.5
Papers:
2.5W
Citations:
6.5W

Organization

T
tsinghua university
Scholars:
11.8W
Papers: 10.0W
Citations: 137
T
Tsinghua Shenzhen International Graduate School
Scholars:
6.8K
Papers: 4.9K
Citations: 9