Caffe学习：使用pycaffe读取caffemodel参数

jiarenyf

于 2015-11-15 20:13:54 发布

阅读量1.9w

点赞数 5

分类专栏： machine-learning caffe python 文章标签： caffe pycaffe

本文链接：https://blog.csdn.net/u011762313/article/details/49851795

版权

本文详细介绍了如何使用PyCaffe库来读取和解析Caffe模型的参数，包括权重和配置信息。通过实例演示了加载预训练模型的过程，为理解和运用Caffe模型提供实践指导。

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

#!/usr/bin/env python

# 引入“咖啡”
import caffe

import numpy as np

# 使输出的参数完全显示
# 若没有这一句，因为参数太多，中间会以省略号“……”的形式代替
np.set_printoptions(threshold='nan')

# deploy文件
MODEL_FILE = 'caffe_deploy.prototxt'
# 预先训练好的caffe模型
PRETRAIN_FILE = 'caffe_iter_10000.caffemodel'

# 保存参数的文件
params_txt = 'params.txt'
pf = open(params_txt, 'w')

# 让caffe

最低0.47元/天解锁文章