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