python调用caffe

首先需要安装caffe for python,安装过程可以参考:Http://blog.csdn.net/u011961856/article/details/76557509

Python 中调用caffe库函数为:

import caffe

如果不再caffe目录下运行汇报错:

import caffe

ImportError: No module named caffe

这是因为找不到caffe库,我们可以在代码中设置caffe路径,例如我的caffe安装目录为/home/program/caffe,则设置路径如下:

import sys
caffe_root='/home/program/caffe'
sys.path.insert(0, caffe_root + '/python')
import caffe