flystem-usls/examples/yolov8
Jamjamjon beda8ef803
Add YOLOv8-OBB and some bug fixes (#9)
* Add YOLOv8-Obb & Refactor outputs

* Update README.md
2024-04-21 17:06:58 +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 Add YOLOv8-OBB and some bug fixes (#9) 2024-04-21 17:06:58 +08:00
main.rs Add YOLOv8-OBB and some bug fixes (#9) 2024-04-21 17:06:58 +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