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

2022-12-02 13:28 开发作者： jdkholmes

生成dll

IDE使用visual studi编程o 2015。

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

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

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

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

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

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

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

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

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

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

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

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

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

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

重新生成解决方案，此时生成的t开发者_Go学习fdll.dll会出现在C#的debug目录下。

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

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

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

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

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

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

以上为个人经验，希望能给大家一个参考，也希望大家多多支持我们。