安装onnx
1、安裝pip【https://blog.csdn.net/quantum7/article/details/83543591】
sudo apt-get install python-pip? ? ? ? ?#python2
sudo apt-get install python3-pip? ? ? #python3
2、caffe安裝(ubantu>=17)【http://caffe.berkeleyvision.org/install_apt.html】
sudo apt install caffe-cpu cpu版本3、caffe2安裝(ubantu18.04)
通過anaconda安裝的
a、先安裝anaconda:【https://www.howtoing.com/how-to-install-anaconda-on-ubuntu-18-04-quickstart】【私密】
不要根據curl -0 那個方法下載,我用curl下載完,找不到下載的東西在哪。我用瀏覽器下的,這時下載的文件在Downloads文件夾下。根據上面的教程就安裝好了。
anaconda教程【https://docs.anaconda.com/anaconda/】
?
b、然后安裝caffe2:【https://caffe2.ai/docs/getting-started.html?platform=ubuntu&configuration=prebuilt#anaconda-install-path】
?創建一個新的環境安裝caffe2
conda install pytorch-nightly-cpu -c pytorchca caffe2的cpu版本4、onnx的安裝? 官方文檔【https://github.com/onnx/onnx】
我使用的文檔【https://blog.csdn.net/mengdeyuanterry/article/details/79750065】
conda install -c conda-forge onnxconda install -c conda-forge protobuf numpypip install onnx?5、caffe2環境有了,onnx也能正常導入了,在根據官方文檔進行測試時【https://github.com/onnx/tutorials/blob/master/tutorials/OnnxCaffe2Import.ipynb】,還是報錯了,如下
? ? ? ? ? ?Import Error: no module named 'past'
解決方法:pip install future? ?【https://www.cnblogs.com/huolifeng/p/6412183.html】
接下來就是下載 squeezenet.onnx【https://github.com/onnx/models/tree/master/squeezenet】
6、為了導入你們倆,把哥哥害苦了,
7、onnx模型轉換成caffe2運行的模型時
convert-onnx-to-caffe2 assets/squeezenet.onnx --output predict_net.pb --init-net-output init_net.pb報錯:ModuleNotFoundError: No module named 'click'
? ? ? ? pip?install click? ? 【https://github.com/alexwlchan/backup_tumblr/issues/10】
注:不要用pip3安裝,pip3安裝時,系統會說已經安裝好了,但是如果用conda list查看時,發現并沒有相關的模塊
總結:這個博客的主要目的是記錄一下由onnx模型轉caffe2模型的探索過程。在ubunatu系統上先由anaconda安裝caffe2,然后再安裝onnx。最后根據文檔進行操作,不過這怎么產生倆個pb文件,pb文件不應該是tensorflow框架產生的嗎,caffe不應該產生.prototxt和.caffemodel文件嗎。。。一山放過一山攔,爬了一坑又一坑
總結
- 上一篇: VIDEOIO ERROR: V4L:
- 下一篇: visio2016中插入大括号