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 *
```