关闭

Caffe学习:pycaffe接口配置

标签: caffe
693人阅读 评论(1) 收藏 举报
分类:

转: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
>>> 
0
1

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:3572次
    • 积分:137
    • 等级:
    • 排名:千里之外
    • 原创:9篇
    • 转载:11篇
    • 译文:0篇
    • 评论:21条
    文章分类
    最新评论