tracel-ai / burn

Burn is a new comprehensive dynamic Deep Learning Framework built using Rust with extreme flexibility, compute efficiency and portability as its primary goals.
https://burn.dev
Apache License 2.0
9.02k stars 447 forks source link

Load `yolov8` network into `burn-rs` #675

Open Luni-4 opened 1 year ago

Luni-4 commented 1 year ago

Ultralytics YOLOv8 is a cutting-edge, state-of-the-art (SOTA) model that builds upon the success of previous YOLO versions and introduces new features and improvements to further boost performance and flexibility. YOLOv8 is designed to be fast, accurate, and easy to use, making it an excellent choice for a wide range of object detection and tracking, instance segmentation, image classification and pose estimation tasks.

yolov8n.zip contains ONNX model

yolov8n onnx

Missing operators:

antimora commented 1 year ago

I am working on a refactor or clean up of ONNX data. After I'll be working on Gather. Reshape should be already available. I checked in ONNX test for it.

Luni-4 commented 1 year ago

Thank you @antimora! Yep, you're right, Reshape is already implemented