C++下如何将TensorFlow模型封装成DLL供C#调用

生成dll

IDE使用visual studio 2015。

首先新建一个C++win32工程,选择dll,包括导出符号,但不要预编译头。

模板会提供一个工程同名的头文件和一个源文件。

在源文件中将TensorFlow头文件和其他需要使用的头文件引用进来。

C++下如何将TensorFlow模型封装成DLL供C#调用_第1张图片

在源文件中定义需要export的函数。

C++下如何将TensorFlow模型封装成DLL供C#调用_第2张图片

在头文件中给出需导出函数的声明。

C++下如何将TensorFlow模型封装成DLL供C#调用_第3张图片

extern 关键字后加“C”保证在C#调用时不生成中间函数名。

编译环境选择X64,配置好include、lib目录以及附加依赖项文件,生成解决方案。

在输出目录会看见生成了dll文件。

使用C#调用dll

new project,创建一个C#控制台程序进行测试,将C#工程添加到当前solution中,右键设为工程第一启动项。打开C#工程属性,将build选项中的平台设置为x64(很关键)

C++下如何将TensorFlow模型封装成DLL供C#调用_第4张图片

在C#的cs文件中首先include 头文件system.runtime.interopservices。将dll工程的输出目录设置为C#的debug目录,与EXE文件同级。

重新生成解决方案,此时生成的tfdll.dll会出现在C#的debug目录下。

使用DllImport导入dll文件,如下图所示代码即可调用dll中的函数。

C++下如何将TensorFlow模型封装成DLL供C#调用_第5张图片

若出现badimage报错,试图加载不正确的dll,可能是由于位数选择出错,dll和C#冲突。

若出现找不到dll报错,可能是由于未将此dll所需的依赖dll一同拷入debug文件夹下。

下图中的TensorFlow的dll即是tfdll的依赖库。

C++下如何将TensorFlow模型封装成DLL供C#调用_第6张图片

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

你可能感兴趣的:(C++下如何将TensorFlow模型封装成DLL供C#调用)