下载caffe源码
源码地址:https://github.com/BVLC/caffe
ImportError:No module named _caffe
可以先运行一下代码,如果报错是“ImportError:No module named _caffe”,那么需要编译caffe源码。
编译之前需要将Makefile.config.example复制为Makefile.config。这里如果使用的opencv版本是3.0以上的,需要将Makefile.config中的
# OPENCV_VERSION := 3
设置为不注释,即
OPENCV_VERSION := 3
然后进行以下步骤:
# 进入caffe目录
cd /path/to/caffe-master
# 清除原编译
(sudo) make clean
# 编译,make -j4或者make -j8,根据具体环境使用
(sudo) make -j4 / (sudo) make -j8
# 编译python caffe环境
(sudo) make pycaffe
# 测试
(sudo) make runtest
测试全部是OK即可。
ImportError:No module named caffe
先运行代码,如果是报错“ImportError:No module named caffe”,那么应该是caffe的路径问题,可以在代码里添加一下内容。
import sys
sys.path.insert(0,caffe_root+'python)
import caffe
这样import caffe就没有问题了。