Skip to content

Latest commit

 

History

History
26 lines (18 loc) · 3.52 KB

README.md

File metadata and controls

26 lines (18 loc) · 3.52 KB

RTMDet

Abstract

Our tech-report will be released soon.

Results and Models

Backbone size SyncBN box AP Params(M) FLOPS(G) TRT-FP16-Latency(ms) Config Download
RTMDet-tiny 640 Yes 40.9 4.8 8.1 0.98 config model | log
RTMDet-s 640 Yes 44.5 8.89 14.8 1.22 config model | log
RTMDet-m 640 Yes 49.1 24.71 39.27 1.62 config model | log
RTMDet-l 640 Yes 51.3 52.3 80.23 2.44 config model | log
RTMDet-x 640 Yes 52.6 94.86 141.67 3.10 config model | log

Note:

  1. The inference speed is measured on an NVIDIA 3090 GPU with TensorRT 8.4.3, cuDNN 8.2.0, FP16, batch size=1, and without NMS.
  2. We still directly use the weights trained by mmdet currently. A re-trained model will be released later.