Hugging Face's logo

eebyp
/
RT-DETR

Object Detection
RT-DETR
real-time
Model card Files
xet
 Community
eebyp commited on
Commit
4fdbc7c
·
verified ·
1 Parent(s): 59d3dd7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +60 -15
README.md CHANGED
@@ -8,23 +8,44 @@ tags:
8
 
9
  # 🎯 RT-DETR
10
 
11
- Official weights for RT-DETR (Real-Time DEtection TRansformer) series — ECCV 2026.
12
 
13
  ## 📋 Versions
14
 
15
- | Version | Directory | Files | Status |
16
- |:--------|:----------|:------|:-------|
17
- | **v4** | `v4/` | S / M / L / X | ✅ Available |
18
- | v1 / v2 / v3 | `v1/`, `v2/`, `v3/` | — | 🔜 Coming soon |
 
 
19
 
20
- ## 📦 Model Files (v4)
21
 
22
- | Model | File | Size |
23
- |:------|:-----|-----:|
24
- | **RT-DETRv4-S** | `v4/RTv4-S-hgnet.pth` | 161.4 MB |
25
- | **RT-DETRv4-M** | `v4/RTv4-M-hgnet.pth` | 303.2 MB |
26
- | **RT-DETRv4-L** | `v4/RTv4-L-hgnet.pth` | 482.0 MB |
27
- | **RT-DETRv4-X** | `v4/RTv4-X-hgnet.pth` | 963.7 MB |
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
28
 
29
  ## 📥 Download
30
 
@@ -37,12 +58,36 @@ hf_hub_download(
37
  filename="v4/RTv4-S-hgnet.pth",
38
  local_dir="./model"
39
  )
 
 
 
 
 
 
 
 
 
 
 
 
 
 
40
  ```
41
 
 
 
 
 
 
 
 
 
 
42
  ## 🔗 Source
43
 
44
- - **Official Repository**: [RT-DETRs/RT-DETRv4](https://github.com/RT-DETRs/RT-DETRv4)
45
- - **Paper**: [DETRs Beat YOLOs on Real-time Object Detection](https://arxiv.org/abs/2304.08069)
 
46
 
47
  ## 📄 License
48
 
@@ -50,4 +95,4 @@ Apache 2.0
50
 
51
  ---
52
 
53
- > **Note:** This repository is hosted for **personal research purposes only**. For production use, please refer to the [official repository](https://github.com/RT-DETRs/RT-DETRv4).
 
8
 
9
  # 🎯 RT-DETR
10
 
11
+ Official weights for RT-DETR (Real-Time DEtection TRansformer) series.
12
 
13
  ## 📋 Versions
14
 
15
+ | Version | Directory | Files | Paper | Status |
16
+ |:--------|:----------|:------|:------|:-------|
17
+ | **v1** | `v1/` | R18 / R50 / R101 (3) | [CVPR 2024](https://arxiv.org/abs/2304.08069) | ✅ Available |
18
+ | **v2** | `v2/` | S / M / L / X (4) | [2024](https://arxiv.org/abs/2407.17140) | ✅ Available |
19
+ | **v3** | `v3/` | — | [WACV 2025](https://arxiv.org/abs/2409.08475) | 🔜 Coming soon |
20
+ | **v4** | `v4/` | S / M / L / X (4) | [ECCV 2026](https://arxiv.org/abs/2507.18455) | ✅ Available |
21
 
22
+ ## 📦 Model Files
23
 
24
+ ### v1 — RT-DETR (CVPR 2024)
25
+
26
+ | Model | File | Size | AP<sup>val</sup> |
27
+ |:------|:-----|-----:|-----------------:|
28
+ | **RT-DETR-R18** | `v1/rtdetr_r18vd_dec3_6x_coco_from_paddle.pth` | 77.2 MB | 46.4 |
29
+ | **RT-DETR-R50** | `v1/rtdetr_r50vd_6x_coco_from_paddle.pth` | 164.4 MB | 53.1 |
30
+ | **RT-DETR-R101** | `v1/rtdetr_r101vd_6x_coco_from_paddle.pth` | 293.3 MB | 54.3 |
31
+
32
+ ### v2 — RT-DETRv2 (2024)
33
+
34
+ | Model | File | Size | AP<sup>val</sup> |
35
+ |:------|:-----|-----:|-----------------:|
36
+ | **RT-DETRv2-S** | `v2/rtdetrv2_r18vd_120e_coco_rerun_48.1.pth` | 77.4 MB | 48.1 |
37
+ | **RT-DETRv2-M** | `v2/rtdetrv2_r50vd_m_7x_coco_ema.pth` | 140.5 MB | 51.9 |
38
+ | **RT-DETRv2-L** | `v2/rtdetrv2_r50vd_6x_coco_ema.pth` | 164.5 MB | 53.4 |
39
+ | **RT-DETRv2-X** | `v2/rtdetrv2_r101vd_6x_coco_from_paddle.pth` | 293.5 MB | 54.3 |
40
+
41
+ ### v4 — RT-DETRv4 (ECCV 2026)
42
+
43
+ | Model | File | Size | AP<sup>val</sup> |
44
+ |:------|:-----|-----:|-----------------:|
45
+ | **RT-DETRv4-S** | `v4/RTv4-S-hgnet.pth` | 161.4 MB | 50.0 |
46
+ | **RT-DETRv4-M** | `v4/RTv4-M-hgnet.pth` | 303.2 MB | 52.5 |
47
+ | **RT-DETRv4-L** | `v4/RTv4-L-hgnet.pth` | 482.0 MB | 53.8 |
48
+ | **RT-DETRv4-X** | `v4/RTv4-X-hgnet.pth` | 963.7 MB | 54.7 |
49
 
50
  ## 📥 Download
51
 
 
58
  filename="v4/RTv4-S-hgnet.pth",
59
  local_dir="./model"
60
  )
61
+
62
+ # Download v2-S
63
+ hf_hub_download(
64
+ repo_id="eebyp/RT-DETR",
65
+ filename="v2/rtdetrv2_r18vd_120e_coco_rerun_48.1.pth",
66
+ local_dir="./model"
67
+ )
68
+
69
+ # Download v1-R50
70
+ hf_hub_download(
71
+ repo_id="eebyp/RT-DETR",
72
+ filename="v1/rtdetr_r50vd_6x_coco_from_paddle.pth",
73
+ local_dir="./model"
74
+ )
75
  ```
76
 
77
+ ## 🔄 Version History
78
+
79
+ | Version | Year | Conference | Backbone | Key Contribution |
80
+ |:--------|:-----|:-----------|:---------|:-----------------|
81
+ | **v1** | 2024 | CVPR 2024 | ResNet | First real-time end-to-end detector |
82
+ | **v2** | 2024 | — | ResNet | Improved baseline with bag-of-freebies |
83
+ | **v3** | 2025 | WACV 2025 | ResNet | Hierarchical dense positive supervision |
84
+ | **v4** | 2026 | ECCV 2026 | HGNetv2 | Vision foundation model distillation |
85
+
86
  ## 🔗 Source
87
 
88
+ - **v1 / v2**: [lyuwenyu/RT-DETR](https://github.com/lyuwenyu/RT-DETR)
89
+ - **v4**: [RT-DETRs/RT-DETRv4](https://github.com/RT-DETRs/RT-DETRv4)
90
+ - **Paper (v1)**: [DETRs Beat YOLOs on Real-time Object Detection](https://arxiv.org/abs/2304.08069)
91
 
92
  ## 📄 License
93
 
 
95
 
96
  ---
97
 
98
+ > **Note:** This repository is hosted for **personal research purposes only**. For production use, please refer to the official repositories above.