YOLO系列模型 pt文件转化为ONNX导出

啥是onnx

在这里插入图片描述
 Microsoft 和合作伙伴社区创建了 ONNX 作为表示机器学习模型的开放标准。许多框架(包括 TensorFlow、PyTorch、scikit-learn、Keras、Chainer、MXNet 和 MATLAB)的模型都可以导出或转换为标准 ONNX 格式。 在模型采用 ONNX 格式后,可以在各种平台和设备上运行这些模型。
 懂了吧,简单来说就是emmmmm微软干了类似统一度量衡的事情,类似Pytorch,Keras,mxnet的模型都可以导出成onnx的格式。之后放到别的平台上去运行,也就是说,一旦导出成了onnx格式,就跟你之前用的是啥库没关系了哈。

怎么导出

from ultralytics import YOLO

# load a pretrained model (recommended for training)
model = YOLO('best.pt')

# Export the model
success = model.export(format='onnx')

print(success)

导出之后

很神奇的onnx模型可视化
在这里插入图片描述
 可视化的样子是这个样子的,可以看到每一个模块的Metadata。在这个metadata里面可以看到输入输出维度,还有原本使用的库函数,就像下面这样(这里拿原本的yolo v8m最后的output0 block为例):
在这里插入图片描述

相关推荐

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-06-07 02:32:03       98 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-06-07 02:32:03       106 阅读
  3. 在Django里面运行非项目文件

    2024-06-07 02:32:03       87 阅读
  4. Python语言-面向对象

    2024-06-07 02:32:03       96 阅读

热门阅读

  1. MySQL学习——从命令行调用MySQL 程序

    2024-06-07 02:32:03       26 阅读
  2. 【视频转码】ZLMediaKit漏洞报告的问题

    2024-06-07 02:32:03       30 阅读
  3. 2024-05-30 vscode python不能调试

    2024-06-07 02:32:03       33 阅读
  4. 【WPF编程宝典】第9讲:几何图形

    2024-06-07 02:32:03       30 阅读
  5. 设计模式之状态模式

    2024-06-07 02:32:03       30 阅读
  6. 展锐8541E配置USB ACM

    2024-06-07 02:32:03       32 阅读
  7. mongodb 编码格式 Detected BSON

    2024-06-07 02:32:03       28 阅读
  8. element-plus datePicker 如何输出时间戳

    2024-06-07 02:32:03       31 阅读
  9. 0098__gconf、dconf系统配置编辑器

    2024-06-07 02:32:03       30 阅读
  10. Python实现 植物大战僵尸

    2024-06-07 02:32:03       31 阅读