如何将通道最后的keras模型转换为通道第一的ONNX模型

Question

我想将 keras .h5 模型转换为 ONNX 模型。但是我的模型最后是训练通道，我希望它首先是通道。 我发现keras2onnx.convert_keras 中有一个名为channel_first_inputs 的参数，但在他们的official site 上找不到任何关于如何使用它的示例。

我正在执行此步骤，作为将我的 keras 模型转换为 .engine 模型的过程的一部分。有没有其他方法可以做到这一点而不需要 onnx 中间步骤？

---

在keras2onnx代码里面搜索参数名，发现如下用法

• 在Here：channel_first_inputs=['input_1']
• 在Here：channel_first_inputs=[model.input_names[0]]

Answer 1

我使用了channel_first_inputs=['input_1']，效果很好。