anonymous-submission-dataset-code commited on
Commit
b81d4be
·
verified ·
1 Parent(s): 6504fcc

Update README.md

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