CoolFace
Modelpublic

anonymous-submission-dataset-code/TiBuDB_trained_weights

sourceHugging Faceetalab-2.0updated 5mo agoView on Hugging Face
0likes
Model Card

Model Weights

The trained weights for all benchmarks are hosted on Hugging Face.

Weights Organization

Download the weights and place them in the TiBuDB_trained_weights/ directory.

TaskModelWeight FileDescriptionSAHI Crop SizeInference Size
DetectionYOLO26xbest_det_yolo26x_seed1000_baseline.ptBaseline (1x)128128
DetectionYOLO26xbest_det_yolo26x_seed1000_x4.ptUpscaled (4x)128512
DetectionRF-DETRbest_ema_det_rfdetr_large_seed0_baseline.pthTransformer Baseline128N/A
SegmentationYOLO26xbest_seg_yolo26x_seed100_baseline.ptBaseline (1x)128128
SegmentationYOLO26xbest_seg_yolo26x_seed100_x4.ptUpscaled (4x)128512
SegmentationRF-DETRbest_ema_seg_rfdetr_large_seed100_baseline.pthTransformer Baseline128N/A
OBBYOLO26xbest_obb_yolo26x_seed5000_baseline.ptOriented Bbox (1x)128128
OBBYOLO26xbest_obb_yolo26x_seed5000_x4.ptOriented Bbox (4x)128512
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")