flystem
/
flystem-usls
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
flystem-usls/examples/yolov9
History
Jamjamjon a5141a53be
Accelerate model pre-processing and post-processing (#23) 
* Add X struct to handle input and preprocessing

*Add Ops struct to manage common operations

* Use SIMD (fast_image_resize) to accelerate model pre-processing and post-processing
 		2024-06-30 15:19:34 +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 Accelerate model pre-processing and post-processing (#23) 2024-06-30 15:19:34 +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