flystem
/
flystem-usls
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
flystem-usls/examples/yolov10
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 feat: support yolov10 (#15) 2024-06-01 16:42:38 +08:00
demo.png feat: support yolov10 (#15) 2024-06-01 16:42:38 +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 yolov10

Export ONNX Model

  • Export

    # clone repo and install dependencies
git clone https://github.com/THU-MIG/yolov10.git
cd yolov10
pip install -r requirements.txt

# donwload `pt` weights
wget https://github.com/THU-MIG/yolov10/releases/download/v1.1/yolov10n.pt

# export ONNX model
yolo export model=yolov10n.pt format=onnx opset=13 simplify dynamic

Results