Packages from Quy Nhon AI
Project description
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).
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
yolov8_onnx-1.0.0.tar.gz
(4.5 kB
view hashes)
Built Distribution
Close
Hashes for yolov8_onnx-1.0.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 5baf4d72aa6ce10887578c9619729f20a348bdcd5f0d5adaeebae559b06e99fb |
|
MD5 | 319df53b4cdffb66b0ae9f6bbb784094 |
|
BLAKE2b-256 | 1cdd54a7591834141e2833c4c4a9adca360b05262770d62ac031722d911989a1 |