python调用caffe

发布时间:2019-09-10 08:56:01编辑:auto阅读(1740)

    首先需要安装caffe for python,安装过程可以参考:http://blog.csdn.net/u011961856/article/details/76557509

    python 中调用caffe库函数为:

    import caffe

    如果不再caffe目录下运行汇报错:

    import caffe

    ImportError: No module named caffe

    这是因为找不到caffe库,我们可以在代码中设置caffe路径,例如我的caffe安装目录为/home/program/caffe,则设置路径如下:

    import sys
    caffe_root='/home/program/caffe'
    sys.path.insert(0, caffe_root + '/python')
    import caffe

    调用成功:


关键字