如何将channel last keras模型转换为channel first ONNX模型

我想把keras.h5模型转换成ONNX模型。但我的模型是最后一个训练频道，我希望它是第一个频道。 我发现

keras2onnx.convert\u keras

中有一个名为

channel\u first\u inputs

的参数，但找不到任何关于如何在其上使用它的示例

我将此步骤作为将keras模型转换为.engine模型过程的一部分。是否有其他方法不需要进行NX中间步骤即可完成此操作

---

在keras2onnx代码中搜索参数名，我发现了以下用法

• 输入：channel\u first\u inputs=['input\u 1']
• 输入：channel\u first\u inputs=[model.input\u names[0]]

---

我使用了

通道第一输入=['input\u 1']

，效果很好