yolov8-onnx 1.0.0

Packages from Quy Nhon AI

YOLOV8 ONNX

Description

This package is compatible with YoloV8 for object detection program, using ONNX format model (CPU speed can be x2 times faster). This code is referenced from this awesome repo.

Installation

You can install the package using pip:

pip install yolov8_onnx

Usage

Step 1: Convert your pre-trained model to ONNX format.

from ultralytics import YOLO

# Load your pre-trained model
model = YOLO('your-trained-model.pt')

# Export the model
model.export(format='onnx', imgsz=640, dynamic=True)

Step 2: Use in your code.

from yolov8_onnx import DetectEngine

engine = DetectEngine(model_path='your-model.onnx', conf_thres=0.5, iou_thres=0.1)

output = engine(image) # cv2 image

License

This project is licensed under the Apache License 2.0 - see the LICENSE file for details.

" I hope this package will be useful for your projects" Nguyễn Trường Lâu (from akaOCR).