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

Deep learning 如何合并两个ONNX深度学习模型,deep-learning,onnx,onnxruntime,Deep Learning,Onnx,Onnxruntime,我有两个ONNX格式的模型。这两个模型相似(都是预先训练的深度学习模型,例如ResNet50模型)。它们之间的唯一区别在于,最后一层针对不同的数据集进行了优化/重新培训 我想合并这两个模型的前k层,如下所示。这将提高推理的性能 为了让我的情况更清楚,这些例子来自于其他机器学习工具来实现这个特性 您可以使用ONNX包及其公开的api()来修改模型/图形。提供了许多更高级别的函数来编辑ONNX图形,包括合并两个子图的功能

我有两个ONNX格式的模型。这两个模型相似(都是预先训练的深度学习模型,例如ResNet50模型)。它们之间的唯一区别在于,最后一层针对不同的数据集进行了优化/重新培训

我想合并这两个模型的前k层,如下所示。这将提高推理的性能

为了让我的情况更清楚,这些例子来自于其他机器学习工具来实现这个特性


您可以使用ONNX包及其公开的api()来修改模型/图形。

提供了许多更高级别的函数来编辑ONNX图形,包括合并两个子图的功能