如何配置pycaffe

首先,使用cmake配置、生成caffe的vs2015工程時,設定生成python接口,即BUILD項->BUILD_python、BUILD_python_layer,注意使用CMake生成VS工程以前必定確保python3.5或2.7已成功安裝(目前caffe僅支持這兩個版本),且numpy庫也已安裝,不然沒法生成VS工程:python

image

而後VS編譯生成python接口:shell

image

其中_caffe.pyd就是接口文件,這個須要編譯生成,有了它意味着pycaffe已經準備好了,能夠使用了,接着須要將其配置到python環境,編輯以下腳本文件:windows

import os.path as osp
import sys


path = "D:\caffe\python"

if path not in sys.path:
    sys.path.insert(0,path)

print(sys.path)

將其保存爲init_pycaffe.py。其中,path是caffe的python接口所在的上一級目錄,即:google

image

注意,path必須是絕對路徑。使用pycaffe時,須要在python shell輸入:spa

import init_pycaffe.py3d

便可導入pycaffe環境,以下圖所示:code

image

import時可能會提示缺乏skimage、google(protobuf)等模塊,只要安裝這些模塊便可(注意,須要先安裝pip):blog

pip install scikit-image接口

pip install protobufip

若是不想每次都要執行init_pycaffe.py,那麼能夠在系統環境變量中增長「PYTHONPATH」,將pycaffe路徑賦值給它亦可,之後再開發pycaffe相關的應用就不用先執行init_pycaffe.py導入caffe模塊了

 

另,windows平臺配置opencv,其實就是控制檯輸入以下指令便可:

pip install opencv-python

pip install opencv-contrib-python

詳情參見https://pypi.org/project/opencv-python/

相關文章
相關標籤/搜索