mirror of https://github.com/open-mmlab/mmyolo.git
bb3aa48689
* add cspnext * add neck * update * align s,tiny test * update * update * fix bug of UT * fix typehint * fix resize bug * update s and tiny * update s and tiny link * update convert * add cls config with rtmdet * update config * fix ci error Co-authored-by: wanghonglie <wanghonglie@pjlab.org.cn> |
||
|---|---|---|
| .. | ||
| cspnext_imagenet_pretrain | ||
| README.md | ||
| metafile.yml | ||
| rtmdet_l_syncbn_8xb32-300e_coco.py | ||
| rtmdet_m_syncbn_8xb32-300e_coco.py | ||
| rtmdet_s_syncbn_8xb32-300e_coco.py | ||
| rtmdet_tiny_syncbn_8xb32-300e_coco.py | ||
| rtmdet_x_syncbn_8xb32-300e_coco.py | ||
README.md
RTMDet
Abstract
Our tech-report will be released soon.
Results and Models
| Backbone | size | SyncBN | ox 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
mmdetcurrently. A re-trained model will be released later.