Android平台的深度學習模型載入

2021-10-07 09:57:11 字數 507 閱讀 7817

經實驗,得到如下結果:

1、android通過opencv的dnn模組,可以載入caffe模型,但載入tf、onnx、pt失敗;vs中通過opencv的dnn模組載入模型,相同結果。

2、centernet的pth模型轉為pt模型,android中通過pytorch.module載入失敗。

3、android中通過pytorch.module載入大小為13.7m、3.84m的.pt模型成功,載入大小為240m、75m的.pt模型失敗;centernet的.pt模型為225m。

4、結論:將pytorch模型轉化為caffe模型,通過opencv的dnn模組載入。

實時性摸底:公開的pytorch分類模型,android端執行通過,ztebv0710手機0.5fps;公開的caffe目標檢測模型,android端執行通過,ztebv0710手機0.1fps。

ztebv0710手機:處理器mt6755,8核cpu,cpu頻率1.8ghz。

搭建深度學習平台記錄

第一次搭建深度學習平台,簡單的記錄一下過程。一 首先我安裝了anaconda。首先因為現在的3.7版本與tf不適用,所以先降級為3.6.8 在降級的時候開啟anaconda prompt,乙個類似命令列的東西,控制anaconda環境python各種庫的安裝。輸入 conda install pyt...

深度學習 Windows平台下的Caffe編譯教程

一 安裝cuda7.5 cuda是英偉達推出的gpu加速運算平台 我這裡安裝的是cuda7.5,已經安裝過的忽略,還沒有安裝過的這裡有安裝教程.windows下面安裝還是非常簡單的.微軟官方也移植了caffe,在windows下面的配置會比較簡單一點.三 開啟caffe master資料夾,然後看到...

阿里雲深度學習平台試玩

python cifar pai.py buckets users kylefan program cifar 10 cifar 10 batches py checkpointdir users kylefan program cifar 10 checkpoint 其中 buckets對應下圖的...