ONNX2Keras：将ONNX模型图转为Keras模型格式
5星
- 浏览量: 0
- 大小:None
- 文件类型：None
简介：
ONNX2Keras是一款工具，能够高效地将ONNX模型转换成Keras支持的模型格式，助力深度学习领域的研究者和开发者便捷使用不同框架训练的模型。
ONNX至Keras深度神经网络转换器使用onnx2keras库实现。该工具需要TensorFlow 2.0环境。
函数定义如下：
```
onnx_to_keras(onnx_model, input_names, input_shapes=None, name_policy=None, verbose=True, change_ordering=False) -> {Keras model}
```
参数包括：
- onnx_model：要转换的ONNX模型。
- input_names：包含图形输入名称的列表。
- input_shapes（可选）：覆盖输入形状（实验性功能）。
- name_policy（可选）：设置为`name_policy`, `short` 或 `default`，用于控制图层名称生成策略（实验性功能）。
- verbose (默认为True) ：是否启用详细输出信息。
- change_ordering: 是否将数据排序方式更改为HWC模式以适应特定需求（实验性功能）。
使用示例：
```
import onnx
from onnx2 import *
```
全部评论 (0)
还没有任何评论哟~
