flystem-usls/examples/yolov9
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
..
README.md Add YOLOv8-OBB and some bug fixes (#9) 2024-04-21 17:06:58 +08:00
demo.png Add YOLOv8-OBB and some bug fixes (#9) 2024-04-21 17:06:58 +08:00
main.rs Add MODNet model (#11) 2024-04-30 15:26:53 +08:00

README.md

Quick Start

cargo run -r --example yolov9

Donwload or Export ONNX Model

  • Download

    yolov9-c-dyn-fp16

  • Export

    # clone repo and install dependencies
    git clone https://github.com/WongKinYiu/yolov9.git
    cd yolov9
    pip install -r requirements.txt
    
    # donwload `pt` weights
    wget https://github.com/WongKinYiu/yolov9/releases/download/v0.1/yolov9-c.pt
    
    # export ONNX model
    python export.py --weights yolov9-c.pt --include onnx --simplify --dynamic
    

Results