flystem-usls/examples/yolov8
Jamjamjon fc970fc117
Update ort and improve the speed of preprocessing
* Add onnx proto

* Update ort to 2.0.0-rc.2

* Improve the speed of resizing

* Fix yolo-seg bug

* Update README.md
2024-05-12 18:12:34 +08:00
..
README.md Add YOLOv8-OBB and some bug fixes (#9) 2024-04-21 17:06:58 +08:00
demo-cls.png Add YOLOv8-OBB and some bug fixes (#9) 2024-04-21 17:06:58 +08:00
demo-det.png Add YOLOv8-OBB and some bug fixes (#9) 2024-04-21 17:06:58 +08:00
demo-obb-2.png Add YOLOv8-OBB and some bug fixes (#9) 2024-04-21 17:06:58 +08:00
demo-obb.png Add YOLOv8-OBB and some bug fixes (#9) 2024-04-21 17:06:58 +08:00
demo-pose.png Add YOLOv8-OBB and some bug fixes (#9) 2024-04-21 17:06:58 +08:00
demo-seg.png Update ort and improve the speed of preprocessing 2024-05-12 18:12:34 +08:00
main.rs Update ort and improve the speed of preprocessing 2024-05-12 18:12:34 +08:00

README.md

Quick Start

cargo run -r --example yolov8

Export YOLOv8 ONNX Models

pip install -U ultralytics

# export onnx model with dynamic shapes
yolo export model=yolov8m.pt format=onnx simplify dynamic
yolo export model=yolov8m-cls.pt format=onnx simplify dynamic
yolo export model=yolov8m-pose.pt format=onnx simplify dynamic
yolo export model=yolov8m-seg.pt format=onnx simplify dynamic
yolo export model=yolov8m-obb.pt format=onnx simplify dynamic

# export onnx model with fixed shapes
yolo export model=yolov8m.pt format=onnx simplify
yolo export model=yolov8m-cls.pt format=onnx simplify
yolo export model=yolov8m-pose.pt format=onnx simplify
yolo export model=yolov8m-seg.pt format=onnx simplify
yolo export model=yolov8m-obb.pt format=onnx simplify

Result

Task Annotated image
Obb img
Instance Segmentation img
Classification img
Detection img
Pose img