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