# RetinaMask: Learning to predict masks improves state-of-the-art single-shot detection for free

Source: [RetinaMask: Learning to predict masks improves state-of-the-art single-shot detection for free](<https://www.sota2.com/research/paper/68fa65381e209131cea09148>)

Collection index: [llms.txt](<https://www.sota2.com/research/llms.txt>)

Data updated: 2026-01-04T03:02:20.621Z

Authors: Cheng-Yang Fu, Mykhailo Shvets, Alexander C. Berg

Published: 2019-01-10T19:35:28.000Z

## Abstract

Recently two-stage detectors have surged ahead of single-shot detectors in the accuracy-vs-speed trade-off. Nevertheless single-shot detectors are immensely popular in embedded vision applications. This paper brings single-shot detectors up to the same level as current two-stage techniques. We do this by improving training for the state-of-the-art single-shot detector, RetinaNet, in three ways: integrating instance mask prediction for the first time, making the loss function adaptive and more stable, and including additional hard examples in training. We call the resulting augmented network RetinaMask. The detection component of RetinaMask has the same computational cost as the original RetinaNet, but is more accurate. COCO test-dev results are up to 41.4 mAP for RetinaMask-101 vs 39.1mAP for RetinaNet-101, while the runtime is the same during evaluation. Adding Group Normalization increases the performance of RetinaMask-101 to 41.7 mAP. Code is at:https://github.com/chengyangfu/retinamask

## Recorded benchmark results

A research result applies to its stated dataset, split, metric and evaluation configuration. Compare like-for-like settings, preserve the metric's unit and direction, and distinguish paper-reported results from independent reproduction. The best result recorded here is not a claim of exhaustive, current global coverage.

- [Instance Segmentation on MS COCO (test-dev)](<https://www.sota2.com/research/sota/instance-segmentation-on-ms-coco-test-dev.md>): Instance Segmentation; MS COCO (test-dev); mAP@\[.5:.95\]: 34.7.

- [Instance Segmentation on COCO 2017 (test-dev)](<https://www.sota2.com/research/sota/instance-segmentation-on-coco-2017-test-dev.md>): Instance Segmentation; COCO 2017 (test-dev); AP (Overall): 34.7.

- [Instance Segmentation on COCO (test-dev)](<https://www.sota2.com/research/sota/instance-segmentation-on-coco-test-dev.md>): Instance Segmentation; COCO (test-dev); APM: 36.7.

- [Object Detection on COCO v2017 (test-dev)](<https://www.sota2.com/research/sota/object-detection-on-coco-v2017-test-dev.md>): Object Detection; COCO v2017 (test-dev); mAP: 42.6.

- [Object Detection on COCO (test-dev)](<https://www.sota2.com/research/sota/object-detection-on-coco-test-dev.md>): Object Detection; COCO (test-dev); mAP: 42.6.

## Original sources and implementation

[Original paper](<https://arxiv.org/abs/1901.03353>)

[PDF](<https://arxiv.org/pdf/1901.03353.pdf>)

[Implementation](<https://github.com/chengyangfu/retinamask>)
