關(guān)于Linux下conda安裝caffe與pb轉(zhuǎn)caffe問(wèn)題
最近碰到了需要將tensorflow的pb模型轉(zhuǎn)為caffe的prototxt格式的問(wèn)題
發(fā)現(xiàn)除了安裝mmdnn以外，還依賴(lài)caffe，配置caffe環(huán)境會(huì)遇到各種坑，因?yàn)槲也恍枰猚affe訓(xùn)練模型，只是為了轉(zhuǎn)換，因此只需要裝caffe-cpu版本即可，最后經(jīng)過(guò)不斷嘗試，在百度的AI Studio上安裝上了caffe，這里記錄一下caffe的安裝：
Anaconda的安裝就不贅述了，百度的AI Studio上已經(jīng)有conda了，因此直接從caffe的安裝開(kāi)始：
1.先新建一個(gè)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.測(cè)試安裝是否成功：
python3import caffe
無(wú)報(bào)錯(cuò)說(shuō)明安裝成功，退出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安裝最簡(jiǎn)單方便！
實(shí)測(cè)，python3.6、python3.5都可以按照上述方法安裝caffe
到此這篇關(guān)于關(guān)于Linux下conda安裝caffe與pb轉(zhuǎn)caffe問(wèn)題的文章就介紹到這了,更多相關(guān)Linux下conda安裝caffe內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
這篇文章主要介紹了Linux中的常見(jiàn)漏洞修復(fù),具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2023-12-12
使用‘fsck’修復(fù)Linux中文件系統(tǒng)錯(cuò)誤的方法
這篇文章主要給大家介紹了關(guān)于如何使用'fsck'修復(fù)Linux中文件系統(tǒng)錯(cuò)誤的相關(guān)資料，文中通過(guò)示例代碼以及圖文介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2019-03-03
Linux工具之Nethogs按進(jìn)程監(jiān)控網(wǎng)絡(luò)帶寬的安裝部署
NetHogs是一個(gè)小型的net top工具,不像大多數(shù)工具那樣拖慢每個(gè)協(xié)議或者是每個(gè)子網(wǎng)的速度而是按照進(jìn)程進(jìn)行帶寬分組，這篇文章主要介紹了Linux工具Nethogs按進(jìn)程監(jiān)控網(wǎng)絡(luò)帶寬，值得收藏,需要的朋友可以參考下2019-07-07
這篇文章主要介紹了Ubuntu 18.04更換apt-get源的方法，本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2020-04-04
Apache環(huán)境下配置多個(gè)ssl證書(shū)搭建多個(gè)站點(diǎn)的方法
這篇文章主要介紹了Apache環(huán)境下配置多個(gè)ssl證書(shū)搭建多個(gè)站點(diǎn)的方法,本文圖文并茂給大家介紹的非常詳細(xì)，具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2018-07-07
Linux在應(yīng)用層讀寫(xiě)寄存器的方法及實(shí)現(xiàn)實(shí)例
這篇文章主要介紹了Linux在應(yīng)用層讀寫(xiě)寄存器的方法及實(shí)現(xiàn)實(shí)例的相關(guān)資料,需要的朋友可以參考下2017-02-02
這篇文章主要介紹了Linux locate命令的使用方法,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-02-02
詳解如何在 Linux 啟動(dòng)時(shí)自動(dòng)執(zhí)行命令或腳本
這篇文章主要介紹了詳解如何在 Linux 啟動(dòng)時(shí)自動(dòng)執(zhí)行命令或腳本，主要有兩種方式，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下。2017-03-03
本篇文章主要介紹了CentOS下tar打包解壓，小編覺(jué)得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2017-01-01
CentOS7環(huán)境下gcc(版本10.2.0)升級(jí)詳細(xì)過(guò)程
大家好，本篇文章主要講的是CentOS7環(huán)境下gcc（版本10.2.0）升級(jí)詳細(xì)過(guò)程，感興趣的同學(xué)快來(lái)看一看吧，希望對(duì)你有幫助2021-11-11