關(guān)于Linux下conda安裝caffe與pb轉(zhuǎn)caffe問題
最近碰到了需要將tensorflow的pb模型轉(zhuǎn)為caffe的prototxt格式的問題
發(fā)現(xiàn)除了安裝mmdnn以外，還依賴caffe，配置caffe環(huán)境會遇到各種坑，因?yàn)槲也恍枰猚affe訓(xùn)練模型，只是為了轉(zhuǎn)換，因此只需要裝caffe-cpu版本即可，最后經(jīng)過不斷嘗試，在百度的AI Studio上安裝上了caffe，這里記錄一下caffe的安裝：
Anaconda的安裝就不贅述了，百度的AI Studio上已經(jīng)有conda了，因此直接從caffe的安裝開始：
1.先新建一個python3.5環(huán)境
conda create -n caffe-py3.5 python=3.5 -c defaults
輸入 y 進(jìn)行安裝
2.激活環(huán)境
安裝完畢后，激活進(jìn)入環(huán)境
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官網(wǎng)下載的mobilenetv1模型為例，轉(zhuǎn)換前先使用netron查看pb模型輸入和輸出節(jié)點(diǎn)name，input shape：
記住name以及input shape
7.進(jìn)行pb ----> caffe轉(zhuǎn)換，命令行輸入：
mmconvert -sf tensorflow -iw mobilenet_v1.pb --inNodeName input --inputShape 192,192,3 --dstNodeName MobilenetV1/Predictions/Reshape_1 -df caffe -om tf_mobilenet
8.轉(zhuǎn)換結(jié)果：
成功轉(zhuǎn)換為caffe模型：
總結(jié):
caffe還是在Linux下用conda安裝最簡單方便！
實(shí)測，python3.6、python3.5都可以按照上述方法安裝caffe
到此這篇關(guān)于關(guān)于Linux下conda安裝caffe與pb轉(zhuǎn)caffe問題的文章就介紹到這了,更多相關(guān)Linux下conda安裝caffe內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
本文介紹了Linux系統(tǒng)中進(jìn)程等待的重要性,包括防止內(nèi)存泄露和獲取子進(jìn)程狀態(tài),詳細(xì)講解了wait()和waitpid()函數(shù)的使用,接著,探討了進(jìn)程替換的概念,通過exec系列函數(shù)實(shí)現(xiàn)程序替換,講解了不同函數(shù)的用法和參數(shù)含義,需要的朋友可以參考下2024-12-12
linux下因?yàn)橄到y(tǒng)編碼問題造成亂碼的快速解決方法
下面小編就為大家?guī)硪黄猯inux下因?yàn)橄到y(tǒng)編碼問題造成亂碼的快速解決方法。小編覺得挺不錯的，現(xiàn)在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2016-10-10
這篇文章主要介紹了Linux中的常見漏洞修復(fù),具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2023-12-12
VMware虛擬機(jī)中Ubuntu18.04無法連接網(wǎng)絡(luò)的解決辦法
這篇文章主要為大家詳細(xì)介紹了VMware虛擬機(jī)中Ubuntu18.04無法連接網(wǎng)絡(luò)的解決辦法，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-03-03
bug分支和feature分支_動力節(jié)點(diǎn)Java學(xué)院整理
這篇文章主要介紹了bug分支和feature分支，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2017-08-08
文章介紹了服務(wù)器安全設(shè)置,包括啟用防火墻、檢查端口、關(guān)閉端口訪問、添加特定IP訪問特定端口的方法步驟,以及注意事項(xiàng)2025-04-04
這篇文章主要介紹了Linux防火墻iptables添加白名單方式，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2023-06-06
linux把一個文件的內(nèi)容復(fù)制到另一個文件的末尾
這篇文章主要介紹了linux把一個文件的內(nèi)容復(fù)制到另一個文件的末尾,文中給大家提到了linux 把文件末尾輸出到另一個文件的方法，需要的朋友可以參考下2019-08-08
ssh報錯no key alg的解決方法(關(guān)于低版本連接高版本ssh)
這篇文章主要介紹了ssh報錯no key alg(關(guān)于低版本連接高版本ssh),通過ssh-keygen命令重新生成ssh主機(jī)秘鑰，可以不用重啟sshd服務(wù)，本文給大家介紹的非常詳細(xì)，需要的朋友可以參考下2023-07-07
這篇文章主要介紹了Linux 配置靜態(tài)IP的方法，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2017-01-01