File size: 1,099 Bytes
15503c4 879e042 15503c4 879e042 216d9e9 879e042 216d9e9 879e042 2d8283c 879e042 | 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 | ---
license: mit
pipeline_tag: object-detection
---
# Paper Model RTDETR Fine-Tuned Weights
This repository contains RT_DETR weights fine-tuned on the synthetic dataset generated with the Paper Model project:
- GitHub: [Paper Model repository](https://github.com/frekkoz3/Paper-Model?utm_source=chatgpt.com)
## Usage
```python
from huggingface_hub import hf_hub_download
weights_path = hf_hub_download(
repo_id="frekko/paper_model_rt_detr",
filename="model_name.pt",
)
```
## Classes
Class mapping:
0. Header
1. Section
2. Section Title
3. Column
4. Banner
5. Footer
## Trained Models
| Model | Dataset Version | Resolution | Epochs | LR |
|---|---|---|---|---|
| `1024_0_v1_rt_detr` | V1 | 1024 | 40 | 0.001 |
| `1024_0_v2_rt_detr` | V2 | 1024 | 60 | 0.001 |
| `1024_1_v2_rt_detr` | V2 | 1024 | 60 | 1e-4 / 1e-5 |
| `1024_0_v3_rt_detr` | V3 | 1024 | 60 | 1e-4 / 1e-5 |
Dataset versions correspond to the synthetic dataset generator versions documented in the Paper Model repository:
- [Paper Model dataset version history](https://github.com/frekkoz3/Paper-Model) |