MixSIS: Mixed-supervised instance segmentation
Organizations: Computer Vision Group, University of Hamburg, Hamburg, Germany
Abstract
Weakly-supervised instance segmentation (WSIS) based on either box-, point-, or image-level annotations has been proposed to overcome the labor-intensive annotation process in fully-supervised instance segmentation. Despite decent results, most WSIS methods only focus on a single annotation level, ignoring that some objects are easy to spot from point annotations, while others require box or even mask annotations. Therefore, we propose the Mixed-Supervised Instance Segmentation (MixSIS) task, which allows a flexible mix of annotation levels (image, point, box, and mask) within a dataset. MixSIS has a high practical relevance, as it allows optimizing the annotation process by annotating challenging objects in more detail, while saving time on simpler instances with less detailed annotations. To address MixSIS, we propose AnnotationsMixer (AnnsMixer) as a baseline. AnnsMixer is a truly mixed-supervised instance segmentation system, which generates pseudo mask annotations from all annotation levels, even mixed within a single image. We evaluate AnnsMixer on both WSIS and MixSIS tasks, largely outperforming state-of-the-art methods. In three case studies, we also demonstrate how the unique features of MixSIS and AnnsMixer can substantially reduce the annotation time (-90%) or improve the results (+34%) in instance segmentation with a simple annotation rule or leveraging heterogeneous data. Our code is available at https://github.com/chwilms/MixSIS
Figures & tables
| System | Mask R-CNN [ 9 ] | MAL [ 17 ] | SAPNet [ 26 ] | BoxLevelSet [ 21 ] | AnnsMixer (our) | |
|---|---|---|---|---|---|---|
| Backbone | R-50 | R-50 | R-50 | R-101 | R-50 | |
| Setup | Supervision | |||||
| Mix 1 (25/25/25/25) | mAP | 0.239 | 0.317 | 0.297 | 0.277 | 0.323 |
| mAP 50 | 0.405 | 0.517 | 0.500 | 0.489 | 0.523 | |
| mAP 75 | 0.249 | 0.335 | 0.304 | 0.277 | 0.344 | |
| Mix 2 (10/20/30/40) | mAP | 0.196 | 0.288 | 0.285 | 0.234 | 0.295 |
| System | Supervision | mAP | mAP 50 | mAP 75 |
|---|---|---|---|---|
| Mask R-CNN | 25% | 0.239 | 0.405 | 0.249 |
| AnnsMixer (our) | 25% +75% | 0.267 | 0.429 | 0.288 |
| AnnsMixer (our) | 25% +75% | 0.298 | 0.494 | 0.313 |
| AnnsMixer (our) | 25% +75% | 0.346 | 0.561 | 0.367 |
| AnnsMixer (our) | 25% +25% +25% +25% | 0.323 | 0.523 | 0.344 |
| System | Backbone | Supervision | mAP | mAP 50 | mAP 75 |
|---|---|---|---|---|---|
| Mask R-CNN [ 9 ] | ResNet-50 | 35.5 | 57.0 | 37.8 | |
| BoxInst [ 24 ] | ResNet-50 | 32.1 | 55.1 | 32.4 | |
| DiscoBox [ 18 ] | ResNet-50 | 32.0 | 53.6 | 32.6 | |
| BoxTeacher [ 5 ] | ResNet-50 | 32.9 | 54.1 | 34.2 | |
| MAL [ 17 ] | ResNet-50 | 35.7 | - | - | |
| BoxSeg [ 16 ] | ResNet-101 | 33.3 | 53.4 | 35.0 |
| Supervision | IoU | Recall | Precision |
|---|---|---|---|
| 70.5 | 84.1% | 95.7% | |
| 56.6 | 62.8% | 66.8% | |
| 27.1 | 29.3% | 48.5% |
| Mask R-CNN | AnnsMixer with Mix efficient | AnnsMixer with Mix random | ||
|---|---|---|---|---|
| Dataset | Metric | |||
| COCO | mAP | 0.366 | 0.347 ( -5.2% ) | 0.325 ( -11.2% ) |
| Ann. time | 650.8 s | 114.7 s ( -82.4% ) | 114.7 s ( -82.4% ) | |
| VOC | mAP | 0.302 | 0.275 ( -8.9% ) | 0.267 ( 11.6% ) |
| Ann. time | 207.8 s | 32.1 s ( -84.6% ) | 32.1 s ( -84.6% ) | |
| UVO | mAP | 0.125 | 0.127 ( +1.6% ) | 0.113 ( -9.6% ) |
| System | Supervision | mAP | mAP 50 | mAP 75 |
|---|---|---|---|---|
| Mask R-CNN | 0.288 | 0.530 | 0.288 | |
| AnnsMixer (our) | + | 0.385 | 0.660 | 0.401 |
| System | Supervision | mAP | mAP 50 | mAP 75 |
|---|---|---|---|---|
| Mask R-CNN | 0.229 | 0.423 | 0.216 | |
| AnnsMixer (our) | + | 0.263 | 0.455 | 0.261 |
| System | Datasets | mAP | mAP 50 | mAP 75 |
|---|---|---|---|---|
| Mask R-CNN | MinneApple ( ) | 0.360 | 0.722 | 0.322 |
| AnnsMixer (our) | MinneApple ( ) + MAD ( ) | 0.367 | 0.723 | 0.330 |
| System | Datasets | mAP | mAP 50 | mAP 75 |
|---|---|---|---|---|
| Mask R-CNN | PASCAL VOC ( ) | 0.294 | 0.513 | 0.306 |
| AnnsMixer (our) | PASCAL VOC ( ) + COCO ( ) | 0.325 | 0.546 | 0.341 |
Instance Segmentation
Instance segmentation identifies each object and assigns it a separate pixel-level mask in images or video.
See all 159 papers