Files
usls/examples/yolo-world
Jamjamjon fc970fc117 Update ort and improve the speed of preprocessing
* Add onnx proto

* Update ort to 2.0.0-rc.2

* Improve the speed of resizing

* Fix yolo-seg bug

* Update README.md
2024-05-12 18:12:34 +08:00
..

Quick Start

cargo run -r --example yolo-world

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
    

Results