cargo run -r --example yolo-world
-
Download
-
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
let options = Options::default()
.with_model("ONNX_PATH"); // <= modify this
cargo run -r --example yolo-world