在 Python 和 Java 之间使用 ONNX 进行训练的模型格式转换答案

【问题标题】：Trained model format conversion with ONNX between Python and Java在 Python 和 Java 之间使用 ONNX 进行训练的模型格式转换
【发布时间】：2021-06-05 06:49:57
【问题描述】：

根据 ONNX 文档，如果我用 Python 训练模型，例如 scikit learn 或 Pytorch，我可以将其保存为 ONNX 格式，然后将其部署到 Java 运行时生产环境中。

有没有可能反其道而行之？例如，用 Java 训练模型，然后将其保存为 ONNX 格式或类似格式并部署到 Python 生产环境中？

【问题讨论】：

如果您无法在 Java 中将模型转换为 ONNX 格式，请重新表述问题和标题（在 Java 中将 ML 模型转换为 ONNX）以获得该问题的解决方案。

标签： java python machine-learning onnx

【解决方案1】：

任何以 ONNX 格式保存的模型都可以使用 onnxruntime 包加载到 Python 生产环境中。因此，如果您能够使用任何语言/框架（在本例中为 Java）转换为 ONNX 格式，您可以查看this 以查看如何将 onnx 模型部署到 Python 环境。

【讨论】：