时间戳

首页 > 解决方案 > 从 TensorFlow 到 ONNX - 保存/冻结图形或转换 pkl？

python - 从 TensorFlow 到 ONNX - 保存/冻结图形或转换 pkl？

问题描述

关于从 TensorFlow 到 ONNX 的最佳方式的新手问题：下面列出的两者之间更好（和/或更简单）的方式是什么？

  1. 冻结/保存网络 --> 存储一个 .pb 文件 --> 将 .pb 转换为 .onnx （我正在努力解决这个问题）

  2. 将 .pkl 快照转换为 .onnx。我还没有找到任何材料，所以欢迎任何提示！

标签： pythontensorflowonnx

解决方案

推荐阅读