Files
usls/examples/yolo-world
Jamjamjon 371a08011f Add MODNet model (#11)
* Add MODNet for portrait matting

* Minor fixes

* Move assets to home directory

* Add colormap

* ci

* Update README.md
2024-04-30 15:26:53 +08:00
..
2024-04-30 15:26:53 +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