File size: 1,563 Bytes
c1a405c |
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 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
---
license: apache-2.0
library_name: libreyolo
tags:
- object-detection
- rt-detr
- transformer
- real-time
- pytorch
datasets:
- coco
---
# LibreYOLO RT-DETRv2-M*
RT-DETRv2-M* (r34vd) - 49.9 AP on COCO
## Model Details
- **Architecture**: RT-DETRv2 (Real-Time Detection Transformer v2)
- **Backbone**: ResNet-34 (r34vd)
- **Framework**: PyTorch
- **License**: Apache 2.0
## Performance
| Model | Dataset | Input Size | AP | AP50 | Params | FPS |
|-------|---------|------------|-----|------|--------|-----|
| RT-DETRv2-M* | COCO | 640 | 49.9 | 67.5 | 31M | 161 |
## Usage
```python
from libreyolo import LIBREYOLO
# Load model
model = LIBREYOLO("librertdetrms.pth", size="ms")
# Run inference
result = model("image.jpg", conf_thres=0.5)
# Access results
print(f"Detected {result['num_detections']} objects")
for box, score, cls in zip(result['boxes'], result['scores'], result['classes']):
print(f" Class {cls}: {score:.2f} @ {box}")
```
## Installation
```bash
pip install libreyolo
```
## Citation
```bibtex
@misc{lv2024rtdetrv2,
title={RT-DETRv2: Improved Baseline with Bag-of-Freebies for Real-Time Detection Transformer},
author={Wenyu Lv and Yian Zhao and Qinyao Chang and Kui Huang and Guanzhong Wang and Yi Liu},
year={2024},
eprint={2407.17140},
archivePrefix={arXiv},
primaryClass={cs.CV}
}
```
## Links
- [LibreYOLO GitHub](https://github.com/Libre-YOLO/libreyolo)
- [RT-DETR Paper](https://arxiv.org/abs/2407.17140)
- [Original RT-DETR Repo](https://github.com/lyuwenyu/RT-DETR)
|