File size: 1,610 Bytes
1866708
e5b2856
 
1866708
 
 
 
e5b2856
1866708
 
 
 
 
e5b2856
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
---
title: RT-DETR
emoji: 🎷
colorFrom: pink
colorTo: yellow
sdk: gradio
sdk_version: 4.14.0
python_version: 3.11.6
app_file: app.py
pinned: false
license: mit
---

## RT-DETR Demo

ONNX exported from official pretrained models on COCO + Objects365

|     Model     |     Dataset     | Input Size | AP<sup>val</sup> | AP<sub>50</sub><sup>val</sup> | #Params(M) | FPS |                                                           checkpoint                                                            |
| :-----------: | :-------------: | :--------: | :--------------: | :---------------------------: | :--------: | :-: | :-----------------------------------------------------------------------------------------------------------------------------: |
|  rtdetr_18vd  | COCO+Objects365 |    640     |       49.0       |             66.5              |     20     | 217 | [url<sup>\*</sup>](https://github.com/lyuwenyu/storage/releases/download/v0.1/rtdetr_r18vd_5x_coco_objects365_from_paddle.pth)  |
| rtdetr_r50vd  | COCO+Objects365 |    640     |       55.2       |             73.4              |     42     | 108 | [url<sup>\*</sup>](https://github.com/lyuwenyu/storage/releases/download/v0.1/rtdetr_r50vd_2x_coco_objects365_from_paddle.pth)  |
| rtdetr_r101vd | COCO+Objects365 |    640     |       56.2       |             74.5              |     76     | 74  | [url<sup>\*</sup>](https://github.com/lyuwenyu/storage/releases/download/v0.1/rtdetr_r101vd_2x_coco_objects365_from_paddle.pth) |

### Run locally

Install dependencies

```sh
poetry install
```

Run server

```sh
poetry run gradio app.py
```