iPython - 无法导入caffe

时间:2017-04-18 03:10:25

标签: python caffe

我正在尝试运行用Python编写的a realtime multi-person pose estimation代码。我下载了代码,然后我通过pip install ipython安装了ipython,之后我通过pip install jupyter安装了Jupyter。现在,我在代码的python目录中获得了ipython notebook所需的工具。

服务器已打开(整齐且全部),我可以运行demo.ipynb文件。它有代码以及几个输出图像来显示应用程序的工作原理。这是输出的屏幕截图:

code screencap

所以我可以看到代码和一些plt.imshow()行以及相应的输出。

但是,当我尝试将第64行更改为:

#test_image = '../sample_image/ski.jpg'
test_image = '../sample_image/upper.jpg'
#test_image = '../sample_image/upper2.jpg'

单击Kernel --> Restart & Run All后,它会读取不同的图像,我收到此错误:

ImportError: No module named caffe

然而,在搜索了如何解决它之后,它似乎并不像pip install caffe那样简单。

我在这里错过了一些简单的东西吗?

1 个答案:

答案 0 :(得分:1)

不幸的是,安装caffe并不像“pip”那么简单......
请按照these installation instructions在您的计算机上安装caffe。

更新

对于ubuntu >= 17.04,您可以

~$ sudo apt install caffe-cpu

~$ sudo apt install caffe-cuda

它也适用于anaconda

~$ conda install -c conda-forge caffe