共計 903 個字符,預計需要花費 3 分鐘才能閱讀完成。
本篇內容介紹了“Linux 下 conda 安裝 caffe 與 pb 轉 caffe 問題如何解決”的有關知識,在實際案例的操作過程中,不少人都會遇到這樣的困境,接下來就讓丸趣 TV 小編帶領大家學習一下如何處理這些情況吧!希望大家仔細閱讀,能夠學有所成!
直接從 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 問題如何解決”的內容就介紹到這里了,感謝大家的閱讀。如果想了解更多行業相關的知識可以關注丸趣 TV 網站,丸趣 TV 小編將為大家輸出更多高質量的實用文章!
向 AI 問一下細節
丸趣 TV 網 – 提供最優質的資源集合!
正文完