利用pycaffe提取caffe model中的参数

最新推荐文章于 2019-01-17 17:00:00 发布

原创

最新推荐文章于 2019-01-17 17:00:00 发布 · 5.8k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#caffe

本文介绍了在Ubuntu 14.04系统下，使用Python2.7和最新版Caffe如何方便地查看和修改Caffe网络模型的参数。通过Python接口，可以轻松获取并检查每一层的参数，这对于理解和调整模型非常有帮助。然而，学习率等特定参数的位置并未在文中直接指出。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

实验平台： ubuntu 14.04, python2.7, caffe当前最新版

有时候我们需要查看、修改caffe的网络参数，苦于caffe本身是c++编写，如果想从c++中提取，明显这个太困难。我们使用python来实现这个，每一步都可以查看参数，所以就容易很多。

import sys
import os

# Add caffe package
caffe_python_dir = "your/caffe-master/python"
sys.path.append(caffe_python_dir)
if not os.path.exists(caffe_python_dir):
    print "python caffe not found."
    exit(-1)
import caffe

import numpy as np
from PIL import Image

# load image, switc