Frizy的博客

不断累积

Caffe学习:pycaffe接口配置

转:http://m.blog.csdn.net/article/details?id=47624119

1.pycaffe介绍:

Caffe提供了python的接口(pycaffe),详见caffe/python文件夹。在python代码中import caffe,可以load models(导入模型),forward and backward(前向、反向迭代), handle IO(数据输入输出),visualize networks(net可视化),instrument model solving(自定义优化方法)。所有的模型数据、计算参数都是可供读写的。
    caffe.Net是主要接口,负责导入数据、校验数据、计算模型的。
    caffe.Classifier用于图像分类。
    caffe.Detector用于图像检测。
    caffe.SGDSolver是solver接口。
    caffe.io处理输入输出,数据预处理。
    caffe.draw可视化net的结构。
    Caffe blobs以numpy ndarrays的形式表示,更加方便、高效。

2.配置pycaffe:

安装依赖库:

$ sudo apt-get install python-numpy python-scipy python-matplotlib python-sklearn python-skimage python-h5py python-protobuf python-leveldb python-networkx python-nose python-pandas python-gflags Cython ipython
$ sudo apt-get install protobuf-c-compiler protobuf-compiler 

编译:

make py 

添加~/caffe/python到$PYTHONPATH:

$ sudo gedit /etc/profile
# 添加: export PYTHONPATH=/path/to/caffe/python:$PYTHONPATH
$ source /etc/profile # 使之生效 

测试是否可以引用:

$ python
Python 2.7.6 (default, Jun 22 2015, 17:58:13) 
[GCC 4.8.2] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> import caffe
>>> 
阅读更多
文章标签: caffe
上一篇Ubuntu 14.04 64bit 安装Caffe(GPU版本)--台式机(笔记本请绕道)
下一篇ubuntu14.04编译caffe
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭