Caffe学习:pycaffe接口配置

版权声明:本文为jiarenyf原创文章,未经允许也可以转载,但是附上链接…… https://blog.csdn.net/u011762313/article/details/47624119

1、pycaffe介绍:

  • 参考链接:Interfaces

  • 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:

  • 详见Ubuntu14.04 安装Caffe(仅CPU)第10点。

    • 安装依赖库:

      $ 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
    • 编译:

      $ cd ~/caffe
      $ make pycaffe
    • 添加~/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
      >>> 

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