C++下如何将TensorFlow模型封装成DLL供C#调用
作者:jdkholmes
这篇文章主要介绍了C++下如何将TensorFlow模型封装成DLL供C#调用问题,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教
生成dll
IDE使用visual studio 2015。
首先新建一个C++win32工程,选择dll,包括导出符号,但不要预编译头。
模板会提供一个工程同名的头文件和一个源文件。
在源文件中将TensorFlow头文件和其他需要使用的头文件引用进来。
在源文件中定义需要export的函数。
在头文件中给出需导出函数的声明。
extern 关键字后加“C”保证在C#调用时不生成中间函数名。
编译环境选择X64,配置好include、lib目录以及附加依赖项文件,生成解决方案。
在输出目录会看见生成了dll文件。
使用C#调用dll
new project,创建一个C#控制台程序进行测试,将C#工程添加到当前solution中,右键设为工程第一启动项。打开C#工程属性,将build选项中的平台设置为x64(很关键)
在C#的cs文件中首先include 头文件system.runtime.interopservices。将dll工程的输出目录设置为C#的debug目录,与EXE文件同级。
重新生成解决方案,此时生成的tfdll.dll会出现在C#的debug目录下。
使用DllImport导入dll文件,如下图所示代码即可调用dll中的函数。
若出现badimage报错,试图加载不正确的dll,可能是由于位数选择出错,dll和C#冲突。
若出现找不到dll报错,可能是由于未将此dll所需的依赖dll一同拷入debug文件夹下。
下图中的TensorFlow的dll即是tfdll的依赖库。
以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。