Xuban commited on
Commit
abc7d5b
·
verified ·
1 Parent(s): 7ab2d8c

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +71 -0
README.md ADDED
@@ -0,0 +1,71 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ library_name: libreyolo
4
+ tags:
5
+ - object-detection
6
+ - rt-detr
7
+ - transformer
8
+ - real-time
9
+ - pytorch
10
+ datasets:
11
+ - coco
12
+ ---
13
+
14
+ # LibreYOLO RT-DETRv2-X
15
+
16
+ RT-DETRv2-X (r101vd) - 54.3 AP on COCO
17
+
18
+ ## Model Details
19
+
20
+ - **Architecture**: RT-DETRv2 (Real-Time Detection Transformer v2)
21
+ - **Backbone**: ResNet-101 (r101vd)
22
+ - **Framework**: PyTorch
23
+ - **License**: Apache 2.0
24
+
25
+ ## Performance
26
+
27
+ | Model | Dataset | Input Size | AP | AP50 | Params | FPS |
28
+ |-------|---------|------------|-----|------|--------|-----|
29
+ | RT-DETRv2-X | COCO | 640 | 54.3 | 72.8 | 76M | 74 |
30
+
31
+ ## Usage
32
+
33
+ ```python
34
+ from libreyolo import LIBREYOLO
35
+
36
+ # Load model
37
+ model = LIBREYOLO("librertdetrx.pth", size="x")
38
+
39
+ # Run inference
40
+ result = model("image.jpg", conf_thres=0.5)
41
+
42
+ # Access results
43
+ print(f"Detected {result['num_detections']} objects")
44
+ for box, score, cls in zip(result['boxes'], result['scores'], result['classes']):
45
+ print(f" Class {cls}: {score:.2f} @ {box}")
46
+ ```
47
+
48
+ ## Installation
49
+
50
+ ```bash
51
+ pip install libreyolo
52
+ ```
53
+
54
+ ## Citation
55
+
56
+ ```bibtex
57
+ @misc{lv2024rtdetrv2,
58
+ title={RT-DETRv2: Improved Baseline with Bag-of-Freebies for Real-Time Detection Transformer},
59
+ author={Wenyu Lv and Yian Zhao and Qinyao Chang and Kui Huang and Guanzhong Wang and Yi Liu},
60
+ year={2024},
61
+ eprint={2407.17140},
62
+ archivePrefix={arXiv},
63
+ primaryClass={cs.CV}
64
+ }
65
+ ```
66
+
67
+ ## Links
68
+
69
+ - [LibreYOLO GitHub](https://github.com/Libre-YOLO/libreyolo)
70
+ - [RT-DETR Paper](https://arxiv.org/abs/2407.17140)
71
+ - [Original RT-DETR Repo](https://github.com/lyuwenyu/RT-DETR)