RTDETRv2-pt

KO | EN

gitlite — search

#javascript #python #hacktoberfest #react #llm #go #android #machine-learning #java #cli #pytorch #ios #mcp #open-source

(★ 10)

RTDETRv2 re-implementation using PyTorch

RTDETRv2-pt 최신버젼 다운로드

최종 버전 다운로드 (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# RTDETRv2-pt

RTDETRv2 re-implementation using PyTorch

### Installation
```
git clone https://github.com/balalofernandez/RTDETRv2-pt.git
cd RTDETRv2-pt
conda create -n rtdetr python=3.10
conda activate rtdetr
pip install -r requirements.txt
```

### Train

* Use your own dataset, returning similar dictionary to the one used in `/RTDETRv2-pt/data/coco_dataset.py`
* Adapt the number of classes in the yml file `[num_classes]` to your number of classes + 1 (90 + 1 = 91 in COCO)
* Modify `train.py` to include your dataset.

### Pretraining

This version uses pretrained versions of Resnet for the backbone. However, base models from the [original repo](https://github.com/lyuwenyu/RT-DETR/tree/main/rtdetrv2_pytorch) should work but don't forget to modify the config file and update the path to the checkpoint in `args`.

| Model | Dataset | Input Size | APval | AP50val | #Params(M) | FPS | config| checkpoint | 
| :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: |:---: |
**RT-DETRv2-S** | COCO | 640 | **48.1** (+1.6) | **65.1** | 20 | 217 | [config](./configs/rtdetrv2/rtdetrv2_r18vd_120e_coco.yml) | [url](https://github.com/lyuwenyu/storage/releases/download/v0.2/rtdetrv2_r18vd_120e_coco_rerun_48.1.pth) |
**RT-DETRv2-M** | COCO | 640 | **49.9** (+1.0) | **67.5** | 31 | 161 | [config](./configs/rtdetrv2/rtdetrv2_r34vd_120e_coco.yml) | [url](https://github.com/lyuwenyu/storage/releases/download/v0.1/rtdetrv2_r34vd_120e_coco_ema.pth)
**RT-DETRv2-M** | COCO | 640 | **51.9** (+0.6) | **69.9** | 36 | 145 | [config](./configs/rtdetrv2/rtdetrv2_r50vd_m_7x_coco.yml) | [url](https://github.com/lyuwenyu/storage/releases/download/v0.1/rtdetrv2_r50vd_m_7x_coco_ema.pth)
**RT-DETRv2-L*** | COCO | 640 | **53.4** (+0.3) | **71.6** | 42 | 108 | [config](./configs/rtdetrv2/rtdetrv2_r50vd_6x_coco.yml) | [url](https://github.com/lyuwenyu/storage/releases/download/v0.1/rtdetrv2_r50vd_6x_coco_ema.pth)
**RT-DETRv2-X** | COCO | 640 | 54.3 | **72.8** (+0.1) | 76 | 74 | [config](./configs/rtdetrv2/rtdetrv2_r101vd_6x_coco.yml) | [url](https://github.com/lyuwenyu/storage/releases/download/v0.1/rtdetrv2_r101vd_6x_coco_from_paddle.pth)

* `*` is the current configuration.

### Results

![Results](https://raw.githubusercontent.com/balalofernandez/RTDETRv2-pt/main/rtdetr_image.png)

![Visitor Count](https://profile-counter.glitch.me/balalofernandez-RTDETRv2-pt/count.svg)

RTDETRv2-pt

RTDETRv2-pt 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

RTDETRv2-pt

RTDETRv2-pt 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

Download README