读取和修改caffemodel文件

最新推荐文章于 2022-06-01 17:03:25 发布

原创

最新推荐文章于 2022-06-01 17:03:25 发布 · 9.7k 阅读

3 ·

CC 4.0 BY-SA版权

这篇博客介绍了如何利用Caffe的Python接口读取和修改caffemodel文件中的权重，以便进行网络结构调整和finetune。通过Jupyter Notebook工具，作者展示了如何查看不同层的维度，并将修改后的权重保存到新的模型中。

使用caffe框架训练后得到的权重通过Google Protobuf来存储为.caffemodel的二进制文件，为了更好地更改网络结构和对已有的caffemodel进行finetune，通常我们需要修改一些参数，去为了更好的适应我们自己设计的网络结构并进行训练
好在caffe的Python接口提供了针对caffemodel文件的修改方法
为了更好地可视化，这里使用了Python的jupyter notebook工具
1、读取caffemodel里的权值

import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline

caffe_root = 'F:/caffe_BLVC_VS2013/caffe-windows/'  # 设置caffe的根目录
import sys
sys.path.insert(0, caffe_root + 'python')

import caffe  #导入caffe
caffe.set_mode_gpu()  #设置为gpu模式