Files
usls/examples/yolo-world
Jamjamjon 51b75e9a21 Add YOLOPv2 & Face-Parsing model (#3)
* Add YOLOP and face parsing model
2024-04-14 15:15:59 +08:00
..
2024-03-29 15:54:24 +08:00
2024-04-06 16:16: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