直接從caffe的安裝開始:
1.先新建一個python3.5環境
conda create -n caffe-py3.5 python=3.5 -c defaults
輸入y 進行安裝
#2.啟動環境
安裝完畢後,啟動進入環境
source activate caffe-py3.5
3.安裝caffe-cpu版本或caffe-gpu版本
conda install -c defaults caffeconda install -c defaults caffe-gpu
4.測試安裝是否成功:
python3import caffe
無錯誤說明安裝成功,退出python
exit()
5.安裝tensorflow:
conda install -c defaults tensorflow==1.15.0
6.安裝mmdnn
pip install mmdnn
6. 我們以tf官網下載的mobilenetv1模型為例,轉換前先使用netron查看pb模型輸入和輸出節點name,input shape:
#記住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.轉換結果:
成功轉換為caffe模型:
以上是Linux下conda安裝caffe與pb轉caffe問題怎麼解決的詳細內容。更多資訊請關注PHP中文網其他相關文章!