基于GPU實現(xiàn)的深度學(xué)習(xí)的數(shù)據(jù)庫
項目組基于深度學(xué)習(xí)實現(xiàn)了視頻風(fēng)格化和人像摳圖的功能,但這是在PC/服務(wù)端上跑的,現(xiàn)在需要移植到移動端,因此需要一個移動端的深度學(xué)習(xí)的計算框架。
同類型的庫
caffe-Android-lib 目前應(yīng)該是最便于集成使用的深度學(xué)習(xí)框架庫。
tensorflow和mxnet據(jù)說也有對應(yīng)的android庫,因時間原因暫未。
CNNdroid,網(wǎng)址,這個是用
render 作優(yōu)化的深度學(xué)習(xí)框架,不過就代碼實現(xiàn)和實際測試結(jié)果來看,性能一般。
工程定位
實現(xiàn)可實時、體積小、通用的深度學(xué)習(xí)預(yù)測框架。
可實時
跟PC或服務(wù)器不同,移動設(shè)備上的GPU可不一定有CPU強悍(多線程+neon/vfp),但在需要實時計算的場景(主要是相機預(yù)覽和視頻播放),往往都是基于OpenGL渲染環(huán)境的。
實時的情況下,深度學(xué)習(xí)框架的輸入和輸出都在GPU端,使用CPU進行計算往往需要拷貝圖像出來,算好后再傳到GPU端,因此基于GPU實現(xiàn)的深度學(xué)習(xí)的庫能持平CPU版本的效率就有足夠優(yōu)勢了。
比如實時摳人像這個case:
對每一幀相機預(yù)覽產(chǎn)生的數(shù)據(jù),系統(tǒng)將其映射為opengl 的一個external texture,然后需要 計算出一個 mask texture,與原先的texture作混合,顯示出來。如果mask texture 的計算在cpu上進行,則需要每幀先把 graphicbuffer 的數(shù)據(jù)拷貝出來,計算出mask后上傳到 mask texture 去,產(chǎn)生一來一回兩次額外拷貝。
通用
本工程需要支持 caffe 產(chǎn)出的模型文件,支持常見的網(wǎng)絡(luò)如lenet、ResNet等等。這個工作量包括編寫相應(yīng)層的算子,設(shè)計網(wǎng)絡(luò)結(jié)構(gòu),解析caffe模型的參數(shù)等。
所幸的是,目前在移動端做好深度學(xué)習(xí)的預(yù)測就足夠了,相比于兼顧訓(xùn)練的結(jié)構(gòu)至少省去2/3的工作量。
非常好我支持^.^
(0) 0%
不好我反對
(0) 0%
下載地址
基于GPU實現(xiàn)的深度學(xué)習(xí)的數(shù)據(jù)庫下載
相關(guān)電子資料下載
- 深度學(xué)習(xí)在工業(yè)缺陷檢測中的應(yīng)用 63
- 探討目前主流3D激光SLAM算法方案 94
- 華為成為中國首個PyTorch基金會Premier會員 453
- 《人工智能在指揮和控制系統(tǒng)中的決策支持》 133
- 中國手機廠商發(fā)力這一領(lǐng)域 外媒:大戲剛剛開始 262
- 百度智能云推出國內(nèi)首個大模型全鏈路生態(tài)支持體系 251
- 人工智能領(lǐng)域存在第一性原理嗎? 53
- 基于機器視覺檢測技術(shù)現(xiàn)狀 41
- 自動駕駛規(guī)控決策方面的建議與解決方案 40
- 里程碑 主線科技無人集卡在舟山港開啟全天候真無人常態(tài)化運營 166