如何合并两个ONNX深度学习模型

Question

我有两个采用ONNX格式的模型。两种模型都是相似的（都是预先训练好的深度学习模型，例如ResNet50模型）。它们之间的唯一区别是最后一层针对不同的数据集进行了优化/重新训练。

我要合并这两个模型的前k个层，如下所示。这样可以提高推理性能。

为使情况更清楚，这些是其他机器学习工具中用于实现此功能Ex的示例。 Pytorch，Keras。

Answer 1

sclblonnx package 提供了许多更高级别的函数来编辑 ONNX 图，包括合并两个子图的能力。

Answer 2

您可以使用ONNX软件包及其公开的API（https://github.com/onnx/onnx/blob/master/docs/PythonAPIOverview.md）对模型/图形进行突变。