首頁 > 運維 > linux運維 > Linux下conda安裝caffe與pb轉caffe問題怎麼解決

Linux下conda安裝caffe與pb轉caffe問題怎麼解決

PHPz
發布: 2023-05-16 14:46:14
轉載
1556 人瀏覽過

直接從caffe的安裝開始:

1.先新建一個python3.5環境

conda create -n caffe-py3.5 python=3.5 -c defaults
登入後複製

輸入y 進行安裝

Linux下conda安裝caffe與pb轉caffe問題怎麼解決

#2.啟動環境

安裝完畢後,啟動進入環境

source activate caffe-py3.5
登入後複製

3.安裝caffe-cpu版本或caffe-gpu版本

conda install -c defaults caffeconda install -c defaults caffe-gpu
登入後複製

4.測試安裝是否成功:

python3import caffe
登入後複製

Linux下conda安裝caffe與pb轉caffe問題怎麼解決

 無錯誤說明安裝成功,退出python

exit()
登入後複製

5.安裝tensorflow:

conda install -c defaults tensorflow==1.15.0
登入後複製

6.安裝mmdnn

pip install mmdnn
登入後複製

Linux下conda安裝caffe與pb轉caffe問題怎麼解決

6. 我們以tf官網下載的mobilenetv1模型為例,轉換前先使用netron查看pb模型輸入和輸出節點name,input shape

Linux下conda安裝caffe與pb轉caffe問題怎麼解決

#記住name以及input shape

7.進行pb ----> caffe轉換,命令列輸入:

mmconvert -sf tensorflow -iw mobilenet_v1.pb --inNodeName input --inputShape 192,192,3 --dstNodeName MobilenetV1/Predictions/Reshape_1 -df caffe -om tf_mobilenet
登入後複製

8.轉換結果:

Linux下conda安裝caffe與pb轉caffe問題怎麼解決

成功轉換為caffe模型:

Linux下conda安裝caffe與pb轉caffe問題怎麼解決

#

以上是Linux下conda安裝caffe與pb轉caffe問題怎麼解決的詳細內容。更多資訊請關注PHP中文網其他相關文章!

相關標籤:
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
最新問題
centos7 - git的linux版本沒有centos的?
來自於 1970-01-01 08:00:00
0
0
0
學習Linux的先行知識
來自於 1970-01-01 08:00:00
0
0
0
Linux下連接資料庫
來自於 1970-01-01 08:00:00
0
0
0
Linux 批次修改檔名
來自於 1970-01-01 08:00:00
0
0
0
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板