burn icon indicating copy to clipboard operation
burn copied to clipboard

Load `yolov8` network into `burn-rs`

Open Luni-4 opened this issue 2 years ago • 7 comments

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:

  • [x] Gather
  • [x] Resize
  • [x] Reshape
  • [x] Split

Luni-4 avatar Aug 22 '23 17:08 Luni-4

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.

antimora avatar Aug 23 '23 03:08 antimora

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

Luni-4 avatar Aug 23 '23 07:08 Luni-4