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

01/10/2019
by   Cheng-Yang Fu, et al.
10

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

READ FULL TEXT

page 5

page 6

page 7

research
03/12/2021

Probabilistic two-stage detection

We develop a probabilistic interpretation of two-stage object detection....
research
04/30/2020

SS3D: Single Shot 3D Object Detector

Single stage deep learning algorithm for 2D object detection was made po...
research
01/18/2018

Extend the shallow part of Single Shot MultiBox Detector via Convolutional Neural Network

Single Shot MultiBox Detector (SSD) is one of the fastest algorithms in ...
research
06/12/2019

Assisted Excitation of Activations: A Learning Technique to Improve Object Detectors

We present a simple and effective learning technique that significantly ...
research
07/26/2023

YOLOBench: Benchmarking Efficient Object Detectors on Embedded Systems

We present YOLOBench, a benchmark comprised of 550+ YOLO-based object de...
research
11/13/2018

Gradient Harmonized Single-stage Detector

Despite the great success of two-stage detectors, single-stage detector ...
research
02/26/2022

Improved Hard Example Mining Approach for Single Shot Object Detectors

Hard example mining methods generally improve the performance of the obj...

Please sign up or login with your details

Forgot password? Click here to reset