flystem-usls/examples/yolov8
Jamjamjon 311d49f5b7
Minor adjustments to YOLO (#17)
2024-06-02 12:49:42 +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 Minor adjustments to YOLO (#17) 2024-06-02 12:49:42 +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