caffe 均值文件binaryproto变为npy文件

最新推荐文章于 2023-07-10 16:53:44 发布

有多少个小土豆

最新推荐文章于 2023-07-10 16:53:44 发布

阅读量302

点赞数

CC 4.0 BY-SA版权

分类专栏： caffe使用

本文链接：https://blog.youkuaiyun.com/htt789/article/details/79757878

caffe使用专栏收录该内容

6 篇文章

订阅专栏

本文介绍如何使用Python脚本将Caffe的二进制均值文件(mean.binaryproto)转换为.npy格式，并计算各通道的平均值。

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

caffe_root = '/home/htt/lisa-caffe-public-lstm_video_deploy/'
import sys
sys.path.insert(0,caffe_root + 'python')
import caffe
import numpy as np
#待转换二进制均值文件
MEAN_PROTO_PATH = '/home/htt/posenet_lmdb/fire/mean/mean.binaryproto'   
#转换生成npy文件            
MEAN_NPY_PATH = 'mean.npy' 
#创建protobuf blob                        
blob = caffe.proto.caffe_pb2.BlobProto()  
#读入mean.binaryproto文件内容         
data = open(MEAN_PROTO_PATH, 'rb' ).read()    
#解析文件内容到blob     
blob.ParseFromString(data) 
#将blob中的均值转换成numpy格式，array的shape （mean_number，channel, hight, width）                        
array = np.array(caffe.io.blobproto_to_array(blob))
#一个array中可以有多组均值存在，故需要通过下标选择其中一组均值
mean_npy = array[0]                                
np.save(MEAN_NPY_PATH ,mean_npy)

mean = np.load(MEAN_NPY_PATH)
a=np.mean(mean[0])
b=np.mean(mean[1])
c=np.mean(mean[2])
print mean
print a,b,c