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

直接從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問題怎么解決

? 版權聲明
THE END
喜歡就支持一下吧
點贊5 分享