如何将通道最后的keras模型转换为通道第一个ONNX模型
How to convert channel last keras model into channel first ONNX model
我想将 keras .h5 模型转换为 ONNX 模型。但是我的模型是最后训练的通道，我希望它首先是通道。
我发现 keras2onnx.convert_keras 中有一个名为 channel_first_inputs 的参数，但找不到关于如何在 official site.
上使用它的任何示例
我正在执行此步骤，作为将我的 keras 模型转换为 .engine 模型的过程的一部分。有没有其他方法可以做到这一点而不需要 onnx 中间步骤？
在keras2onnx代码里面搜索参数名，找到如下用法
我用了channel_first_inputs=['input_1']，效果很好。
我想将 keras .h5 模型转换为 ONNX 模型。但是我的模型是最后训练的通道，我希望它首先是通道。
我发现 keras2onnx.convert_keras 中有一个名为 channel_first_inputs 的参数，但找不到关于如何在 official site.
我正在执行此步骤，作为将我的 keras 模型转换为 .engine 模型的过程的一部分。有没有其他方法可以做到这一点而不需要 onnx 中间步骤？
在keras2onnx代码里面搜索参数名，找到如下用法
我用了channel_first_inputs=['input_1']，效果很好。