CornerNet-Lite: Efficient Keypoint Based Object Detection
Hei Law, Yun Teng, Olga Russakovsky, Jia Deng
Code Available — Be the first to reproduce this paper.
ReproduceCode
- github.com/princeton-vl/CornerNet-LiteOfficialpytorch★ 0
- github.com/PaddlePaddle/PaddleDetectionpaddle★ 14,132
- github.com/takooctopus/CornerNet-Lite-Takopytorch★ 0
- github.com/hollyprince/ObjectsDetection-CornerNet-Litepytorch★ 0
- github.com/jason-su/UCGNetpytorch★ 0
- github.com/Arno3165229/CornerNet_Traffic_Lightpytorch★ 0
Abstract
Keypoint-based methods are a relatively new paradigm in object detection, eliminating the need for anchor boxes and offering a simplified detection framework. Keypoint-based CornerNet achieves state of the art accuracy among single-stage detectors. However, this accuracy comes at high processing cost. In this work, we tackle the problem of efficient keypoint-based object detection and introduce CornerNet-Lite. CornerNet-Lite is a combination of two efficient variants of CornerNet: CornerNet-Saccade, which uses an attention mechanism to eliminate the need for exhaustively processing all pixels of the image, and CornerNet-Squeeze, which introduces a new compact backbone architecture. Together these two variants address the two critical use cases in efficient object detection: improving efficiency without sacrificing accuracy, and improving accuracy at real-time efficiency. CornerNet-Saccade is suitable for offline processing, improving the efficiency of CornerNet by 6.0x and the AP by 1.0% on COCO. CornerNet-Squeeze is suitable for real-time detection, improving both the efficiency and accuracy of the popular real-time detector YOLOv3 (34.4% AP at 30ms for CornerNet-Squeeze compared to 33.0% AP at 39ms for YOLOv3 on COCO). Together these contributions for the first time reveal the potential of keypoint-based detection to be useful for applications requiring processing efficiency.
Tasks
Benchmark Results
| Dataset | Model | Metric | Claimed | Verified | Status |
|---|---|---|---|---|---|
| COCO minival | CornerNet-Saccade (Hourglass-54) | box AP | 42.6 | — | Unverified |
| COCO minival | CornerNet-Saccade (Hourglass-104) | box AP | 41.4 | — | Unverified |
| COCO test-dev | CornerNet-Saccade (Hourglass-104, multi-scale) | box mAP | 43.2 | — | Unverified |