anonymous-submission-dataset-code commited on
Commit
0e9b511
·
verified ·
1 Parent(s): 195a2a9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +48 -3
README.md CHANGED
@@ -1,3 +1,48 @@
1
- ---
2
- license: etalab-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+
3
+ license: etalab-2.0
4
+
5
+
6
+ ## 📦 Model Weights
7
+
8
+
9
+ The trained weights for all benchmarks are hosted on Hugging Face.
10
+
11
+
12
+ ### 📂 Weights Organization
13
+
14
+ Download the weights and place them in the `TiBuDB_trained_weights/` directory.
15
+
16
+
17
+ | Task | Model | Weight File | Description | Sahi crop size | Inference size
18
+
19
+ | :--- | :--- | :--- | :--- | :--- | :--- |
20
+
21
+ | **Detection** | YOLO26x | `best_det_yolo26x_seed1000_baseline.pt` | Baseline (1x) | 128 | 128 |
22
+
23
+ | **Detection** | YOLO26x | `best_det_yolo26x_seed1000_x4.pt` | Upscaled (4x) | 128 | 512 |
24
+
25
+ | **Detection** | RF-DETR | `best_ema_det_rfdetr_large_seed0_baseline.pth` | Transformer Baseline | 128 (basesize) | N/A |
26
+
27
+ | **Segmentation**| YOLO26x | `best_seg_yolo26x_seed100_baseline.pt` | Baseline (1x) | 128 | 128 |
28
+
29
+ | **Segmentation**| YOLO26x | `best_seg_yolo26x_seed100_x4.pt` | Upscaled (4x) | 128 | 512 |
30
+
31
+ | **Segmentation**| RF-DETR | `best_ema_seg_rfdetr_large_seed100_baseline.pth` | Transformer Baseline | 128 (basesize) | N/A |
32
+
33
+ | **OBB** | YOLO26x | `best_obb_yolo26x_seed5000_baseline.pt` | Oriented Bbox (1x) | 128 | 512 |
34
+
35
+ | **OBB** | YOLO26x | `best_obb_yolo26x_seed5000_x4.pt` | Oriented Bbox (4x) | 128 | 512 |
36
+
37
+
38
+ ### 🛠️ Quick Load Example (Ultralytics)
39
+
40
+ ```python
41
+
42
+ from ultralytics import YOLO
43
+
44
+ model = YOLO("TiBuDB_trained_weights/best_det_yolo26x_seed1000_baseline.pt")
45
+
46
+ results = model.predict("data/tibudb_dataset/yolo_det/val/images/test_image.png")
47
+
48
+ ---