【目标检测】YOLOV8实战入门(六)模型导出


export模式用于将YOLOv8模型导出为可用于部署的格式。在此模式下,模型转换为可供其他软件应用程序或硬件设备使用的格式。此模式在将模型部署到生产环境时很有用。

model = YOLO('yolov8n.pt')  
model = YOLO('path/to/best.pt')  

# 导出模型
model.export(format='onnx')

YOLOv8模型的导出设置是指用于保存或导出模型以在其他环境或平台中使用的各种配置和选项。这些设置会影响模型的性能、大小以及与不同系统的兼容性。一些常见导出设置包括导出模型文件的格式(例如ONNX、TensorFlow SavedModel)、模型将在其上运行的设备(例如CPU、GPU)以及每个框存在掩码或多个标签等附加功能。其他可能影响导出过程的因素包括模型正在用于的特定任务以及目标环境或平台的要求或约束。仔细考虑和配置这些设置以确保导出的模型针对预期用例进行了优化并可以在目标环境中有效使用,这一点很重要。

相关参数如下

KeyValueDescription
format'torchscript'format to export to
imgsz640image size as scalar or (h, w) list, i.e. (640, 480)
kerasFalseuse Keras for TF SavedModel export
optimizeFalseTorchScript: optimize for mobile
halfFalseFP16 quantization
int8FalseINT8 quantization
dynamicFalseONNX/TF/TensorRT: dynamic axes
simplifyFalseONNX: simplify model
opsetNoneONNX: opset version (optional, defaults to latest)
workspace4TensorRT: workspace size (GB)
nmsFalseCoreML: add NMS

支持的导出格式如下

Formatformat ArgumentModelMetadata
PyTorch-yolov8n.pt
TorchScripttorchscriptyolov8n.torchscript
ONNXonnxyolov8n.onnx
OpenVINOopenvinoyolov8n_openvino_model/
TensorRTengineyolov8n.engine
CoreMLcoremlyolov8n.mlmodel
TF SavedModelsaved_modelyolov8n_saved_model/
TF GraphDefpbyolov8n.pb
TF Litetfliteyolov8n.tflite
TF Edge TPUedgetpuyolov8n_edgetpu.tflite
TF.jstfjsyolov8n_web_model/
PaddlePaddlepaddleyolov8n_paddle_model/

  • 4
    点赞
  • 24
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

卖报的大地主

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值