flystem-usls/examples/yolov8
Jamjamjon 51b75e9a21
Add YOLOPv2 & Face-Parsing model (#3)
* Add YOLOP and face parsing model
2024-04-14 15:15:59 +08:00
..
README.md Add YOLOPv2 & Face-Parsing model (#3) 2024-04-14 15:15:59 +08:00
demo-cls.jpg Dev (#1) 2024-04-06 16:16:53 +08:00
demo-det.png Add YOLOPv2 & Face-Parsing model (#3) 2024-04-14 15:15:59 +08:00
demo-pose.png Add YOLOPv2 & Face-Parsing model (#3) 2024-04-14 15:15:59 +08:00
demo-seg.png Add YOLOPv2 & Face-Parsing model (#3) 2024-04-14 15:15:59 +08:00
main.rs Add YOLOPv2 & Face-Parsing model (#3) 2024-04-14 15:15:59 +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

# 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

Result

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