flystem-usls/examples/yolo-world

Quick Start

cargo run -r --example yolo-world

Or you can manully

1. Donwload or Export ONNX Model

• Download

  yolov8s-world-v2-shoes

• Or generate your own yolo-world model and then Export

  • Installation

  pip install -U ultralytics
  

  • Generate

  from ultralytics import YOLO
  
  # Initialize a YOLO-World model
  model = YOLO('yolov8m-worldv2.pt')
  
  # Define custom classes
  model.set_classes(["shoes"])
  
  # Save the model with the defined offline vocabulary
  model.save("custom_yolov8m-world-v2.pt")
  

  • Export

  yolo export model=custom_yolov8m-world-v2.pt format=onnx simplify dynamic
  

2. Specify the ONNX model path in main.rs

let options = Options::default()
    .with_model("ONNX_PATH");   // <= modify this

3. Then, run

cargo run -r --example yolo-world

Results