Zecheng Li, Zening Zeng, Yuqi Liang, Jin-Gang Yu
Weakly supervised instance segmentation (WSIS) using only image-level labels is a challenging task due to the difficulty of aligning coarse annotations with the finer task. However, with the advancement of deep neural networks (DNNs), WSIS has garnered significant attention. Following a proposal-based paradigm, we encounter a redundant segmentation problem resulting from a single instance being represented by multiple proposals. For example, we feed a picture of a dog and proposals into the network and expect to output only one proposal containing a dog, but the network outputs multiple proposals. To address this problem, we propose a novel approach for WSIS that focuses on the online refinement of complete instances through the use of MaskIoU heads to predict the integrity scores of proposals and a Complete Instances Mining (CIM) strategy to explicitly model the redundant segmentation problem and generate refined pseudo labels. Our approach allows the network to become aware of multiple instances and complete instances, and we further improve its robustness through the incorporation of an Anti-noise strategy. Empirical evaluations on the PASCAL VOC 2012 and MS COCO datasets demonstrate that our method achieves state-of-the-art performance with a notable margin. Our implementation will be made available at https://github.com/ZechengLi19/CIM.
| Task | Dataset | Metric | Value | Model |
|---|---|---|---|---|
| Weakly-supervised instance segmentation | PASCAL VOC 2012 val | mAP@0.25 | 68.7 | CIM + Mask R-CNN |
| Weakly-supervised instance segmentation | PASCAL VOC 2012 val | mAP@0.5 | 55.9 | CIM + Mask R-CNN |
| Weakly-supervised instance segmentation | PASCAL VOC 2012 val | mAP@0.75 | 30.9 | CIM + Mask R-CNN |
| Instance Segmentation | PASCAL VOC 2012 val | mAP@0.25 | 68.7 | CIM + Mask R-CNN |
| Instance Segmentation | PASCAL VOC 2012 val | mAP@0.5 | 55.9 | CIM + Mask R-CNN |
| Instance Segmentation | PASCAL VOC 2012 val | mAP@0.7 | 37.1 | CIM + Mask R-CNN |
| Instance Segmentation | PASCAL VOC 2012 val | mAP@0.75 | 30.9 | CIM + Mask R-CNN |
| Instance Segmentation | PASCAL VOC 2012 val | mAP@0.25 | 64.9 | CIM |
| Instance Segmentation | PASCAL VOC 2012 val | mAP@0.5 | 51.1 | CIM |
| Instance Segmentation | PASCAL VOC 2012 val | mAP@0.7 | 32.4 | CIM |
| Instance Segmentation | PASCAL VOC 2012 val | mAP@0.75 | 26.1 | CIM |
| Instance Segmentation | COCO 2017 val | AP | 17 | CIM + Mask R-CNN |
| Instance Segmentation | COCO 2017 val | AP@50 | 29.4 | CIM + Mask R-CNN |
| Instance Segmentation | COCO 2017 val | AP@75 | 17 | CIM + Mask R-CNN |
| Instance Segmentation | COCO 2017 val | AP | 11.9 | CIM |
| Instance Segmentation | COCO 2017 val | AP@50 | 22.8 | CIM |
| Instance Segmentation | COCO 2017 val | AP@75 | 11.1 | CIM |
| Instance Segmentation | COCO test-dev | AP | 17.2 | CIM + Mask R-CNN |
| Instance Segmentation | COCO test-dev | AP@50 | 29.7 | CIM + Mask R-CNN |
| Instance Segmentation | COCO test-dev | AP@75 | 17.3 | CIM + Mask R-CNN |
| Instance Segmentation | COCO test-dev | AP | 12 | CIM |
| Instance Segmentation | COCO test-dev | AP@50 | 23 | CIM |
| Instance Segmentation | COCO test-dev | AP@75 | 11.3 | CIM |
| Instance Segmentation | PASCAL VOC 2012 val | mAP@0.25 | 67.8 | CIM + Mask R-CNN |
| Instance Segmentation | PASCAL VOC 2012 val | mAP@0.5 | 55.5 | CIM + Mask R-CNN |
| Instance Segmentation | PASCAL VOC 2012 val | mAP@0.7 | 36.6 | CIM + Mask R-CNN |
| Instance Segmentation | PASCAL VOC 2012 val | mAP@0.75 | 31.1 | CIM + Mask R-CNN |