caffe binaryproto 与 npy相互转换的实例讲解
在caffe中，如果使用的是c++接口，均值文件默认为.binaryproto格式，而如果使用的是python接口，均值文件默认的是numpy的.npy格式，在工作中有时需要将两者进行互相转换，具体方式如下:
binaryproto->npy
importnumpyasnp
importcaffe
importsys
blob=caffe.proto.caffe_pb2.BlobProto()
data=open('mean.binaryproto','rb').read()
blob.ParseFromString(data)
arr=np.array(caffe.io.blobproto_to_array(blob))
out=arr[0]
np.save('mean.npy',out)
npy->binaryproto
data_meanis1*H*W
importnumpyasnp
importcaffe
importsys
blob=caffe.proto.caffe_pb2.BlobProto()
withopen('mean.npy','rb')asf:
data_mean=numpy.load(f)
blob.channels=1
blob.height=data_mean.shape[0]
blob.width=data_mean.shape[1]
blob.data.extend(data_mean.astype(float).flat)
binaryproto_file=open('mean.binaryproto','wb')
binaryproto_file.write(blob.SerializeToString())
binaryproto_file.close()
以上这篇caffebinaryproto与npy相互转换的实例讲解就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持毛票票。
声明：本文内容来源于网络，版权归原作者所有，内容由互联网用户自发贡献自行上传，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任。如果您发现有涉嫌版权的内容，欢迎发送邮件至：czq8825#qq.com（发邮件时，请将#更换为@）进行举报，并提供相关证据，一经查实，本站将立刻删除涉嫌侵权内容。