C++下如何將TensorFlow模型封裝成DLL供C#調(diào)用
生成dll
IDE使用visual studio 2015。
首先新建一個(gè)C++win32工程,選擇dll,包括導(dǎo)出符號,但不要預(yù)編譯頭。
模板會提供一個(gè)工程同名的頭文件和一個(gè)源文件。
在源文件中將TensorFlow頭文件和其他需要使用的頭文件引用進(jìn)來。
在源文件中定義需要export的函數(shù)。
在頭文件中給出需導(dǎo)出函數(shù)的聲明。
extern 關(guān)鍵字后加“C”保證在C#調(diào)用時(shí)不生成中間函數(shù)名。
編譯環(huán)境選擇X64,配置好include、lib目錄以及附加依賴項(xiàng)文件,生成解決方案。
在輸出目錄會看見生成了dll文件。
使用C#調(diào)用dll
new project,創(chuàng)建一個(gè)C#控制臺程序進(jìn)行測試,將C#工程添加到當(dāng)前solution中,右鍵設(shè)為工程第一啟動(dòng)項(xiàng)。打開C#工程屬性,將build選項(xiàng)中的平臺設(shè)置為x64(很關(guān)鍵)
在C#的cs文件中首先include 頭文件system.runtime.interopservices。將dll工程的輸出目錄設(shè)置為C#的debug目錄,與EXE文件同級。
重新生成解決方案,此時(shí)生成的tfdll.dll會出現(xiàn)在C#的debug目錄下。
使用DllImport導(dǎo)入dll文件,如下圖所示代碼即可調(diào)用dll中的函數(shù)。
若出現(xiàn)badimage報(bào)錯(cuò),試圖加載不正確的dll,可能是由于位數(shù)選擇出錯(cuò),dll和C#沖突。
若出現(xiàn)找不到dll報(bào)錯(cuò),可能是由于未將此dll所需的依賴dll一同拷入debug文件夾下。
下圖中的TensorFlow的dll即是tfdll的依賴庫。
以上為個(gè)人經(jīng)驗(yàn),希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
C語言中QString與QByteArray互相轉(zhuǎn)換的方法
本文主要介紹了C語言中QString與QByteArray互相轉(zhuǎn)換的方法,文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-05-05