久久精品人人爽,华人av在线,亚洲性视频网站,欧美专区一二三

Linux下conda安裝caffe與pb轉caffe問題如何解決

151次閱讀
沒有評論

共計 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 網 – 提供最優質的資源集合!

正文完
 
丸趣
版權聲明:本站原創文章,由 丸趣 2024-02-04發表,共計903字。
轉載說明:除特殊說明外本站除技術相關以外文章皆由網絡搜集發布,轉載請注明出處。
評論(沒有評論)
主站蜘蛛池模板: 沾益县| 娄底市| 南投县| 米泉市| 中牟县| 水富县| 江陵县| 富裕县| 苏尼特右旗| 大荔县| 渝中区| 临清市| 南汇区| 云龙县| 广德县| 阜平县| 江川县| 磐石市| 潞西市| 施甸县| 福贡县| 孙吴县| 芜湖市| 奈曼旗| 安陆市| 宁波市| 永仁县| 紫金县| 濮阳市| 民乐县| 石棉县| 信阳市| 资中县| 江孜县| 苍溪县| 托里县| 南澳县| 东台市| 桂东县| 临汾市| 黑龙江省|