您好,欢迎来到爱够旅游网。
搜索
您的当前位置:首页YOLO_V8推理和模型格式转换

YOLO_V8推理和模型格式转换

来源:爱够旅游网

1、推理

设置好task、mode、model和测试图片路径source即可。

task: "detect" # choices=['detect', 'segment', 'classify', 'init'] # init is a special case. Specify task to run.
mode: "predict" # choices=['train', 'val', 'predict', 'export'] # mode to run task in.
model: E:\\DLTest\\YOLOv8\\runs\\detect\\best.pt 
source: MaskDataSet/test/images/ # source directory for images or videos

然后执行:\YOLOv8\ultralytics\yolo\v8\detect\predict.py即可输出模型推理结果。

2、模型格式转换

此处以导出ONNX格式为例,主要的几个参数为:

task: "detect" # choices=['detect', 'segment', 'classify', 'init'] # init is a special case. Specify task to run.
mode: "export" # choices=['train', 'val', 'predict', 'export'] # mode to run task in.
model: E:\\DLTest\\YOLOv8\\runs\\detect\\best.pt
format: onnx #torchscript # format to export to
opset: 12  # ONNX: opset version

然后执行:\YOLOv8\ultralytics\yolo\engine\exporter.py 即可输出格式转换之后的模型。

导出的onnx模型和.pt格式的模型在同一路径下。

可能出现的错误:

错误:ONNX: export failure 0.1s: Unsupported ONNX opset version: 17

原因:onnx opset 设置版本不对,修改版本即可:opset: 12 # ONNX: opset version

因篇幅问题不能全部显示,请点此查看更多更全内容

Copyright © 2019- igbc.cn 版权所有 湘ICP备2023023988号-5

违法及侵权请联系:TEL:199 1889 7713 E-MAIL:2724546146@qq.com

本站由北京市万商天勤律师事务所王兴未律师提供法律服务