将具有多个网络的pytorch模型转换为onnx

文章/答案/技术大牛

发布

3回答

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

deep-learning、onnx、onnxruntime

我有两个ONNX格式的模型。这两个模型是相似的(都是预训练的深度学习模型，例如。ResNet50模型)。它们之间唯一的区别是，最后一层针对不同的数据集进行了优化/重新训练。我想合并这两个模型的前k层，如下所示。这应该会增强推理的性能。

浏览 27提问于2020-08-07得票数 1

3回答

如何将OneHot包含在来自PyTorch的ONNX中

pytorch、one-hot-encoding、onnx、vespa

我正在使用PyTorch训练神经网络并将它们输出到ONNX中.我在索引中使用这些模型，该索引通过TensorRT加载ONNX。我需要一些特性的一个热编码，但这真的很难在Vespa框架内实现。是否有可能在我的ONNX网络中(例如，在网络表示之前)嵌入对某些给定功能的单热编码？如果是这样的话，我应该如何基于PyTorch模型来实现这一点？我已经注意到两件事： <e

浏览 17提问于2021-03-09得票数 3

回答已采纳

1回答

这里我已经得到了我的.onnx格式文件：retinaface.onnx，我想把它转换成PyTorch手机支持的格式：.ptl或.pt，然后我就可以在安卓平台上做推断了。但我没有转换或找到相关的问题。want to load directly but failed mRetinafaceDector = LiteModuleLoader.load(mModelPath + "/retinaface.onnx"); 看起来LiteModule

浏览 278提问于2021-08-28得票数 0

1回答

将PyTorch转换为ONNX模型会增加ALBert的文件大小

python、pytorch、onnx、quantization、onnxruntime

目标:使用这个对albert-base-v2模型执行量化。 PyTorch

浏览 35提问于2022-01-20得票数 0

回答已采纳

2回答

如果没有安装，我怎么才能使用预培训过的模型？

pytorch

我只想在没有安装整个包的情况下使用预先训练过的模型。我能不能把模型模组复制到pytorch？

浏览 1提问于2019-02-26得票数 2

回答已采纳

1回答

deep-learning、pytorch、onnx

我正在尝试将具有多个网络的pytorch模型转换为ONNX，但遇到一些问题。以下是经过训练的模型处理图像的方式： def gen_encode(self, x_a, age_a, age_b=0, training=False, target_age=0):, style_params_b, skip_1, skip_2) return x

浏览 53提问于2021-06-29得票数 3

1回答

如何使用pytorch 0.4.1模型初始化pytorch 0.2.0？

deep-learning、pytorch、pycaffe

我遇到了一个问题，我在pytorch 0.4.1中训练我的模型，但我找不到一个工具来将它转换为caffe模型。如何使用init pytorch 0.2.0的pytorch 0.4.1 model 或者如何将pytorch 0.4.1 model转换为caffe model

浏览 12提问于2018-09-12得票数 0

3回答

如何将.onnx转换为tflite？

tensorflow、keras、deep-learning、pytorch、onnx

我已经通过以下方式将我的模型导出到ONNX：torch_out = torch.onnx.export_params=True) # store the trained parameter weights inside the model file 现在，我试图将模型转换为不幸的是，PyTorch/Caffe2 2对Android的支持相当缺乏或太复杂，但是

浏览 3提问于2018-11-07得票数 20

1回答

-> Onnx -> Tensorflow转换中的张量格式问题

tensorflow、pytorch、onnx、vitis-ai

我有一个关于Tensorflow模型的问题，这个模型是从Pytorch -> Onnx -> Tensorflow转换而来的。问题是，转换后的Tensorflow模型期望输入的格式为Pytorch格式，即(批次大小、数字通道、高度、宽度)，而不是Tensorflow格式(批大小、高度、宽度、数字通道)。因此，我不能使用该模型进一步处理与Vitis AI。因此，我想问一下，是否有任何方法

浏览 8提问于2020-12-17得票数 0

回答已采纳

2回答

python-3.x、subprocess、inference、onnx、tensorrt

我有一个python程序，在.py文件中有以下代码片段，它使用trtexec将ONNX模型转换为TRT引擎： subprocess.run([sys.executable, "-c", "trtexec --onnx=rmpx.onnx --saveEngine=rmpx_engine_pytorch.trt --explicitBatch --fp16"]) subproce

浏览 6提问于2021-06-13得票数 0

2回答

如何将.pth模型转换为.pb文件？

tensorflow、pytorch

我已经通过使用pytorch获得了完整的模型，但是我想将.pth文件转换为.pb，以便在Tensorflow中使用。有谁有什么想法吗？

浏览 3提问于2019-12-23得票数 7

1回答

CNTK C# API支持ConvTranspose吗？

c#、convolution、cntk

我有一个定制的dnn模型，它是我在PyTorch中定义和训练的，我试图将它部署到一个通过ONNX用C#编写的应用程序中。加载模型时，会出现以下错误：经过一些测试后，我发现只有当我将网络的上采样端(它使用转置卷积)返回时才会发生这种情况。

浏览 0提问于2018-09-18得票数 0

回答已采纳

1回答

在ONNX模型转换中未注册架构

tensorflow、tensorflow.js、onnx

我正在尝试将我的pytorch模型转换为tensorflow模型，以便与tensorflowJS一起运行。我使用以下代码将onnx模型转换为tensorflow模型-from onnx_tf.backend import prepareonnx_model = onnx.load/input/<

浏览 88提问于2021-02-20得票数 2

回答已采纳

2回答

如何在浏览器上运行pytorch模型？

javascript、python、web-applications、onnx

我想知道在浏览器中运行对象检测模型的选项是什么，直到现在我已经找到了下面的选项： Onnx:似乎是一个很好的选择，但是我不太明白如何在JS中运行模型，因为我一直只使用python和py火炬，学习路线图是什么？与组件的关系是什么？NCNN:这是我知道的最少的方法，它似乎可以工作在

浏览 14提问于2021-08-20得票数 6

1回答

PyTorch模型中的ONNX对象，不导出

pytorch、onnx、pytorch-lightning、onnxruntime、onnx-coreml

是否可以在不导出的情况下将PyTorch模型转换为ONNX，并进一步将其作为ONNX对象直接在脚本中使用。

浏览 18提问于2021-06-17得票数 0

点击加载更多

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

如何将OneHot包含在来自PyTorch的ONNX中

如何将onnx格式转换为ptl格式？

将PyTorch转换为ONNX模型会增加ALBert的文件大小

如果没有安装，我怎么才能使用预培训过的模型？