關(guān)于Linux下conda安裝caffe與pb轉(zhuǎn)caffe問題
最近碰到了需要將tensorflow的pb模型轉(zhuǎn)為caffe的prototxt格式的問題
發(fā)現(xiàn)除了安裝mmdnn以外，還依賴caffe，配置caffe環(huán)境會遇到各種坑，因為我不需要caffe訓(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é)點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安裝最簡單方便！
實測，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中的常見漏洞修復(fù),具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2023-12-12
使用‘fsck’修復(fù)Linux中文件系統(tǒng)錯誤的方法
這篇文章主要給大家介紹了關(guān)于如何使用'fsck'修復(fù)Linux中文件系統(tǒng)錯誤的相關(guān)資料，文中通過示例代碼以及圖文介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面來一起學(xué)習(xí)學(xué)習(xí)吧2019-03-03
Linux工具之Nethogs按進(jìn)程監(jiān)控網(wǎng)絡(luò)帶寬的安裝部署
NetHogs是一個小型的net top工具,不像大多數(shù)工具那樣拖慢每個協(xié)議或者是每個子網(wǎng)的速度而是按照進(jìn)程進(jìn)行帶寬分組，這篇文章主要介紹了Linux工具Nethogs按進(jìn)程監(jiān)控網(wǎng)絡(luò)帶寬，值得收藏,需要的朋友可以參考下2019-07-07
這篇文章主要介紹了Ubuntu 18.04更換apt-get源的方法，本文給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值,需要的朋友可以參考下2020-04-04
Apache環(huán)境下配置多個ssl證書搭建多個站點的方法
這篇文章主要介紹了Apache環(huán)境下配置多個ssl證書搭建多個站點的方法,本文圖文并茂給大家介紹的非常詳細(xì)，具有一定的參考借鑒價值，需要的朋友可以參考下2018-07-07
Linux在應(yīng)用層讀寫寄存器的方法及實現(xiàn)實例
這篇文章主要介紹了Linux在應(yīng)用層讀寫寄存器的方法及實現(xiàn)實例的相關(guān)資料,需要的朋友可以參考下2017-02-02
這篇文章主要介紹了Linux locate命令的使用方法,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-02-02
這篇文章主要介紹了詳解如何在 Linux 啟動時自動執(zhí)行命令或腳本，主要有兩種方式，具有一定的參考價值，感興趣的小伙伴們可以參考一下。2017-03-03
本篇文章主要介紹了CentOS下tar打包解壓，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2017-01-01
CentOS7環(huán)境下gcc(版本10.2.0)升級詳細(xì)過程
大家好，本篇文章主要講的是CentOS7環(huán)境下gcc（版本10.2.0）升級詳細(xì)過程，感興趣的同學(xué)快來看一看吧，希望對你有幫助2021-11-11