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

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +42 -21
README.md CHANGED
@@ -1,48 +1,69 @@
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
- ---
 
 
 
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
+
32
+ The trained weights for all benchmarks are hosted on [Hugging Face](https://huggingface.co/datasets/anonymous-submission-dataset-code/TiBuDB).
33
+
34
+ ### Weights Organization
35
+
36
+ Download the weights and place them in the `TiBuDB_trained_weights/` directory.
37
+
38
+ | Task | Model | Weight File | Description | SAHI Crop Size | Inference Size |
39
+ | :--- | :--- | :--- | :--- | :--- | :--- |
40
+ | **Detection** | YOLO26x | `best_det_yolo26x_seed1000_baseline.pt` | Baseline (1x) | 128 | 128 |
41
+ | **Detection** | YOLO26x | `best_det_yolo26x_seed1000_x4.pt` | Upscaled (4x) | 128 | 512 |
42
+ | **Detection** | RF-DETR | `best_ema_det_rfdetr_large_seed0_baseline.pth` | Transformer Baseline | 128 | N/A |
43
+ | **Segmentation** | YOLO26x | `best_seg_yolo26x_seed100_baseline.pt` | Baseline (1x) | 128 | 128 |
44
+ | **Segmentation** | YOLO26x | `best_seg_yolo26x_seed100_x4.pt` | Upscaled (4x) | 128 | 512 |
45
+ | **Segmentation** | RF-DETR | `best_ema_seg_rfdetr_large_seed100_baseline.pth` | Transformer Baseline | 128 | N/A |
46
+ | **OBB** | YOLO26x | `best_obb_yolo26x_seed5000_baseline.pt` | Oriented Bbox (1x) | 128 | 128 |
47
+ | **OBB** | YOLO26x | `best_obb_yolo26x_seed5000_x4.pt` | Oriented Bbox (4x) | 128 | 512 |
48
+
49
+ > **Note:** RF-DETR processes images at the native crop size (128) without upscaling; inference size is not applicable.
50
+
51
+ ### Quick Load Example
52
 
53
+ **Ultralytics (YOLO / RT-DETR)**
54
+
55
+ ```python
56
  from ultralytics import YOLO
57
 
58
  model = YOLO("TiBuDB_trained_weights/best_det_yolo26x_seed1000_baseline.pt")
59
+ results = model.predict("path/to/image.png")
60
+ ```
61
 
62
+ **RF-DETR**
63
+
64
+ ```python
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
+ ---