Caffe学习：使用pycaffe读取mean.binaryproto文件参数

最新推荐文章于 2020-12-09 08:38:22 发布

jiarenyf

最新推荐文章于 2020-12-09 08:38:22 发布

阅读量1w

点赞数

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

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

版权

本文详细介绍了如何使用pycaffe库读取Caffe模型中的mean.binaryproto文件，帮助理解该文件参数及其在深度学习模型预处理中的作用。

摘要由CSDN通过智能技术生成

#!/usr/bin/env python
# 引入“咖啡”
import caffe

import numpy as np

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

# 均值文件
MEAN_FILE = 'mean.binaryproto'
# 保存参数的文件
means_txt = 'means.txt'
mf = open(means_txt, 'w')

# 将均值文件读入blob中
mean_blob = caffe.proto.caffe_pb2.BlobProto()
mean_blob.ParseFromString(open(MEAN_FILE, 'rb').rea