Our tech-report will be released soon.
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:
- 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.
- We still directly use the weights trained by
mmdet
currently. A re-trained model will be released later.