---
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)