我们能不能运行一种进口到火把的onnx型号？

Question

我一直在尝试从onnx格式导入一个模型，以便与py手电一起工作。我发现很难找到同样的例子。因为互联网上的大部分资源都在讨论如何将Py手电模型导出到onnx。

我发现torch.onnx()只能导出模型，导入方法还没有实现。直接安装onnx库，可以帮助我执行onnx.load("model_name.onnx")。我该如何使用这种型号的火把？我无法用model.to将模型移动到GPU (device=“GPU”)

Answer 1

PyTorch目前不支持导入onnx模型。在写这个答案的时候，它是一个开放特征请求。

虽然不能保证工作正常，但一个潜在的解决方案是使用微软开发的一个名为MMdnn的工具(不，它不仅仅是窗口！)它支持各种框架之间的转换。不幸的是，onnx只能是转换的目标，而不是源。也就是说，您的可能可以将您的模型导入到另一个框架中，然后使用MMdnn将该框架转换为pytorch。显然，这并不理想，成功的潜力将取决于其他框架如何使用onnx，而onnx可能不适合MMdnn的工作方式。

2022年8月更新

不幸的是，特性请求似乎被拒绝了，Mmdnn也被放弃了。最近有一些第三方工具提供了一些将onnx导入到onnx2pytorch和龙骨火把这样的pytorch中的功能。这两种工具似乎都没有得到积极开发，尽管在这一点上pytorch和onnx相对稳定，因此希望这些工具在今后仍然适用(海事组织的官方支持会更好)。请注意，这两个工具都有未解决的问题，因此，如果其中一个不适合您，则可能有必要同时尝试这两个工具。

2022年9月更新

基于@DanNissenbaum的评论，有一个较新的第三方工具onnx2torch正在积极开发和维护。