溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
本篇內容介紹了“Linux下conda安裝caffe與pb轉caffe問題如何解決”的有關知識，在實際案例的操作過程中，不少人都會遇到這樣的困境，接下來就讓小編帶領大家學習一下如何處理這些情況吧！希望大家仔細閱讀，能夠學有所成！
直接從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問題如何解決”的內容就介紹到這里了，感謝大家的閱讀。如果想了解更多行業相關的知識可以關注億速云網站，小編將為大家輸出更多高質量的實用文章！
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
