HyperAIHyperAI

Command Palette

Search for a command to run...

Console
6 months ago

RTMDet: An Empirical Study of Designing Real-Time Object Detectors

Chengqi Lyu Wenwei Zhang Haian Huang Yue Zhou Yudong Wang Yanyi Liu Shilong Zhang Kai Chen

RTMDet: An Empirical Study of Designing Real-Time Object Detectors

Abstract

In this paper, we aim to design an efficient real-time object detector that exceeds the YOLO series and is easily extensible for many object recognition tasks such as instance segmentation and rotated object detection. To obtain a more efficient model architecture, we explore an architecture that has compatible capacities in the backbone and neck, constructed by a basic building block that consists of large-kernel depth-wise convolutions. We further introduce soft labels when calculating matching costs in the dynamic label assignment to improve accuracy. Together with better training techniques, the resulting object detector, named RTMDet, achieves 52.8% AP on COCO with 300+ FPS on an NVIDIA 3090 GPU, outperforming the current mainstream industrial detectors. RTMDet achieves the best parameter-accuracy trade-off with tiny/small/medium/large/extra-large model sizes for various application scenarios, and obtains new state-of-the-art performance on real-time instance segmentation and rotated object detection. We hope the experimental results can provide new insights into designing versatile real-time object detectors for many object recognition tasks. Code and models are released at https://github.com/open-mmlab/mmdetection/tree/3.x/configs/rtmdet.

Code Repositories

yxb-nku/strip-r-cnn
pytorch
Mentioned in GitHub
cszzshi/SimD
pytorch
Mentioned in GitHub
HVision-NKU/Strip-R-CNN
pytorch
Mentioned in GitHub
open-mmlab/mmyolo
pytorch
Mentioned in GitHub
open-mmlab/mmrotate
pytorch
Mentioned in GitHub
yuyi1005/point2rbox-mmrotate
pytorch
Mentioned in GitHub
fiveai/MoCaE
pytorch
Mentioned in GitHub
RangiLyu/mmdetection_test
pytorch
Mentioned in GitHub
V3Det/mmdetection-V3Det
pytorch
Mentioned in GitHub
open-edge-platform/geti
pytorch
Mentioned in GitHub

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp