代码之家 › 专栏 › 技术社区 › user3452134

加载ONX CNTK C++引发异常

cntk

user3452134 · 技术社区 · 6 年前

我有一个在matlab中创建的文件。我用得很好 python加载:

import cntk as C
z = C.Function.load("Net.onnx", format=C.ModelFormat.ONNX)

在C++中我有例外 Selected CPU as the process wide default device.

将引发异常:

'gemm:无效的形状,输入a和b应为rank=2 矩阵

我用的是进口的Nuget: CNTK.CPUOnly CNTK.Deps.MKL CNTK.Deps.OpenCV.Zip

 #include <stdio.h>
 #include "CNTKLibrary.h"

 void main(){

     std::wstring modelFile(L"Net.onnx");

     //line crash
     CNTK::FunctionPtr modelFunc = CNTK::Function::Load(modelFile, CNTK::DeviceDescriptor::CPUDevice(), CNTK::ModelFormat::ONNX);
}

1 回复 | 直到 6 年前

user3452134 6 年前

最后我在Python中保存了其他的解决方案来模拟CNTK,而不是从C++加载它。以cntk格式(其中原始模型从matlab导出到onnx long way)

python代码

import cntk as C
z = C.Function.load("Net.onnx", format=C.ModelFormat.ONNX)
z.save(os.path.join("folder", "net" + ".dnn"))

C++加载

#include "CNTKLibrary.h"
std::wstring modelFile(L"net.dnn");

CNTK::FunctionPtr modelFunc = CNTK::Function::Load(modelFile, CNTK::DeviceDescriptor::CPUDevice());

推荐文章

Tullhead · 如何在CNTK中使用挤压网?

7 年前

FlyingTeller AEgo · 带有python的CNTK-每个层的激活

7 年前

Frank Ibem · 如何在CNTK模型中访问未命名节点

7 年前

Pablo Gonzalez · CNTK c#逻辑回归w和b变量值

7 年前

Rafael · CNTK运行时错误

7 年前

SJR3t2 · 我如何创建一个致密的numpy层。带CNTK的浮点64?

7 年前

Skiminok · 在CNTK中注意实现Seq2Seq时,轴存在多个问题

7 年前

yingxing.bao · cntk支持下图的哪种神经网络模型?[已关闭]

7 年前

J. Rock · 在GPU上使用简单的矩阵向量积对Theano和CNTK进行基准测试

7 年前

Jonzha · CNTK:模型。不深入0的节点的克隆()

7 年前