首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往

ONNX,一个模型互操作的 Python 库!

<p><span style='font-family: 微软雅黑, "Microsoft YaHei";'>ONNX 是个挺有意思的玩意儿,它就像是不同深度学习框架之间的一个翻译官。想象一下,你用 PyTorch 训练了个模型,但是你的朋友只会用

TensorFlow。这时候 ONNX 就能派上用场了,它能把你的模型翻译成一种通用语言,让不同框架都能听懂。</span></p><br/>

<p><span style='font-family: 微软雅黑, "Microsoft YaHei";'>ONNX 全称是 Open Neural Network Exchange,翻译过来就是开放神经网络交换格式。它是个开源项目,由微软和脸书(现在叫

Meta)一起搞出来的。简单来说,ONNX 就是个中间人,负责在不同的深度学习框架之间传话。    </span></p><br/>

          pre::after {

      position: absolute;

      bottom: 5px;

      right: 10px;

      font-size: 12px;

      color: #999;

      opacity: 0.7;

          }

      </style><pre style="

      border: 1px solid #ddd;

      color: #ce834a;

      page-break-inside: avoid;

      font-family: monospace;

      font-size: 15px;

      line-height: 1.6;

      margin-bottom: 1.6em;

      max-width: 100%;

      overflow: auto;

      padding: 1em 1.5em;

      display: block;

      word-wrap: break-word;

      position: relative;

  "><span></span><code>import onnx</code>

          pre::after {

      position: absolute;

      bottom: 5px;

      right: 10px;

      font-size: 12px;

      color: #999;

      opacity: 0.7;

          }

      </style><pre style="

      border: 1px solid #ddd;

      color: #ce834a;

      page-break-inside: avoid;

      font-family: monospace;

      font-size: 15px;

      line-height: 1.6;

      margin-bottom: 1.6em;

      max-width: 100%;

      overflow: auto;

      padding: 1em 1.5em;

      display: block;

      word-wrap: break-word;

      position: relative;

  "><span></span><code>import torch</code>

          pre::after {

      position: absolute;

      bottom: 5px;

      right: 10px;

      font-size: 12px;

      color: #999;

      opacity: 0.7;

          }

      </style><pre style="

      border: 1px solid #ddd;

      color: #ce834a;

      page-break-inside: avoid;

      font-family: monospace;

      font-size: 15px;

      line-height: 1.6;

      margin-bottom: 1.6em;

      max-width: 100%;

      overflow: auto;

      padding: 1em 1.5em;

      display: block;

      word-wrap: break-word;

      position: relative;

  "><span></span><code>import onnxruntime as ort</code>

<p>温馨提示:ONNX Runtime 和 ONNX 是两个不同的包,别搞混了。你需要单独安装 ONNX Runtime:<code>pip install

onnxruntime</code>。</p><br/>

<p><span style='font-family: 微软雅黑, "Microsoft YaHei";'>import onnx

from onnx import optimizer</span></p><br/>

<p><span style='font-family: 微软雅黑, "Microsoft YaHei";'>ONNX

是个挺有用的工具,特别是当你需要在不同框架之间切换,或者要把模型部署到各种平台上的时候。虽然有些小坑,但只要你注意一下,用起来还是很顺手的。希望这篇文章能帮你对

ONNX 有个基本的了解。要是还有不明白的地方,就多试试,多练练,慢慢就懂了。</span></p><br/>

  • 发表于:
  • 原文链接https://page.om.qq.com/page/OHZc3zzzFPJlTh_eO3SUorRA0
  • 腾讯「腾讯云开发者社区」是腾讯内容开放平台帐号(企鹅号)传播渠道之一,根据《腾讯内容开放平台服务协议》转载发布内容。
  • 如有侵权,请联系 cloudcommunity@tencent.com 删除。

扫码

添加站长 进交流群

领取专属 10元无门槛券

私享最新 技术干货

扫码加入开发者社群
领券