File size: 1,780 Bytes
6504fcc 0e9b511 6504fcc 0e9b511 6504fcc 0e9b511 6504fcc 0e9b511 6504fcc 0e9b511 6504fcc b81d4be | 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 | ---
license: etalab-2.0
---
## Model Weights
The trained weights for all benchmarks are hosted on [Hugging Face](https://huggingface.co/datasets/anonymous-submission-dataset-code/TiBuDB).
### Weights Organization
Download the weights and place them in the `TiBuDB_trained_weights/` directory.
| Task | Model | Weight File | Description | SAHI Crop Size | Inference Size |
| :--- | :--- | :--- | :--- | :--- | :--- |
| **Detection** | YOLO26x | `best_det_yolo26x_seed1000_baseline.pt` | Baseline (1x) | 128 | 128 |
| **Detection** | YOLO26x | `best_det_yolo26x_seed1000_x4.pt` | Upscaled (4x) | 128 | 512 |
| **Detection** | RF-DETR | `best_ema_det_rfdetr_large_seed0_baseline.pth` | Transformer Baseline | 128 | N/A |
| **Segmentation** | YOLO26x | `best_seg_yolo26x_seed100_baseline.pt` | Baseline (1x) | 128 | 128 |
| **Segmentation** | YOLO26x | `best_seg_yolo26x_seed100_x4.pt` | Upscaled (4x) | 128 | 512 |
| **Segmentation** | RF-DETR | `best_ema_seg_rfdetr_large_seed100_baseline.pth` | Transformer Baseline | 128 | N/A |
| **OBB** | YOLO26x | `best_obb_yolo26x_seed5000_baseline.pt` | Oriented Bbox (1x) | 128 | 128 |
| **OBB** | YOLO26x | `best_obb_yolo26x_seed5000_x4.pt` | Oriented Bbox (4x) | 128 | 512 |
> **Note:** RF-DETR processes images at the native crop size (128) without upscaling; inference size is not applicable.
### Quick Load Example
**Ultralytics (YOLO / RT-DETR)**
```python
from ultralytics import YOLO
model = YOLO("TiBuDB_trained_weights/best_det_yolo26x_seed1000_baseline.pt")
results = model.predict("path/to/image.png")
```
**RF-DETR**
```python
from rfdetr import RFDETRLarge
model = RFDETRLarge(pretrain_weights="TiBuDB_trained_weights/best_ema_det_rfdetr_large_seed0_baseline.pth")
results = model.predict("path/to/image.png")
``` |