NeFr25's picture
Rename weights to weights/readme.md
4b90434 verified

🧠 TTA-S2R: Pretrained YOLO Models Collection

This repository contains the pretrained models used in the TTA-S2R (Tidal Turbine Assembly – Sim2Real) pipeline.
All models were trained and evaluated as part of the study on sim-to-real transfer for industrial assembly object detection.

The collection includes YOLOv8, YOLOv9, and YOLO-World models trained under three regimes:

  • Controlled only – trained purely on real controlled data captured with a collaborative robot.
  • Synthetic only – trained purely on domain-randomized synthetic data.
  • Synthetic + Controlled – fine-tuned using both sources for improved sim-to-real generalization.

🧩 Usage

You can load any model directly in Python using the Ultralytics framework:

from ultralytics import YOLO, YOLOWorld

# Example for YOLOv8 or YOLOv9
model = YOLO("best_model_8_synthetic_controlled.pt")

# Example for YOLO-World
model = YOLOWorld("best_model_yoloworld_synthetic_controlled.pt")

results = model.predict(source="path/to/image_or_video.jpg")