直接從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模型:
? 版權聲明
文章版權歸作者所有,未經允許請勿轉載。
THE END