NeFr25 commited on
Commit
4b90434
·
verified ·
1 Parent(s): 7b7ab77

Rename weights to weights/readme.md

Browse files
Files changed (2) hide show
  1. weights +0 -1
  2. weights/readme.md +27 -0
weights DELETED
@@ -1 +0,0 @@
1
- readme.md
 
 
weights/readme.md ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # 🧠 TTA-S2R: Pretrained YOLO Models Collection
2
+
3
+ This repository contains the pretrained models used in the **TTA-S2R (Tidal Turbine Assembly – Sim2Real)** pipeline.
4
+ All models were trained and evaluated as part of the study on **sim-to-real transfer for industrial assembly object detection**.
5
+
6
+ The collection includes YOLOv8, YOLOv9, and YOLO-World models trained under three regimes:
7
+ - **Controlled only** – trained purely on real controlled data captured with a collaborative robot.
8
+ - **Synthetic only** – trained purely on domain-randomized synthetic data.
9
+ - **Synthetic + Controlled** – fine-tuned using both sources for improved sim-to-real generalization.
10
+
11
+ ---
12
+
13
+ ## 🧩 Usage
14
+
15
+ You can load any model directly in Python using the [Ultralytics](https://github.com/ultralytics/ultralytics) framework:
16
+
17
+ ```python
18
+ from ultralytics import YOLO, YOLOWorld
19
+
20
+ # Example for YOLOv8 or YOLOv9
21
+ model = YOLO("best_model_8_synthetic_controlled.pt")
22
+
23
+ # Example for YOLO-World
24
+ model = YOLOWorld("best_model_yoloworld_synthetic_controlled.pt")
25
+
26
+ results = model.predict(source="path/to/image_or_video.jpg")
27
+