caffe binaryproto 与 npy相互转换的实例讲解
更新时间：2018年07月09日 09:12:59 作者：无眠栀
今天小编就为大家分享一篇caffe binaryproto 与 npy相互转换的实例讲解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
在caffe中，如果使用的是c++接口，均值文件默认为.binaryproto格式，而如果使用的是python接口，均值文件默认的是numpy的.npy格式，在工作中有时需要将两者进行互相转换，具体方式如下:
binaryproto -> npy
import numpy as np import caffe import sys 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_mean is 1*H*W
import numpy as np
import caffe
import sys
blob = caffe.proto.caffe_pb2.BlobProto()
with open('mean.npy','rb') as f:
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()
以上这篇caffe binaryproto 与 npy相互转换的实例讲解就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
您可能感兴趣的文章:
相关文章
这篇文章主要为大家详细介绍了pandas中Timestamp类用法，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2017-12-12
这篇文章主要介绍了Python3 使用pillow库生成随机验证码,需要的朋友可以参考下2019-08-08
这篇文章主要为大家详细介绍了TensorFlow实现卷积神经网络CNN，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-03-03
这篇文章主要为大家介绍了Python 异步之生成器示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2023-03-03
本文主要介绍了thon使用js2py库运行js代码，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2022-05-05
Django2.1集成xadmin管理后台所遇到的错误集锦(填坑)
这篇文章主要介绍了Django2.1集成xadmin管理后台所遇到的错误集锦(填坑),小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2018-12-12
这篇文章主要介绍了django美化后台django-suit的安装配置操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-07-07
这篇文章主要介绍了Python使用lambda表达式对字典排序操作,结合实例形式分析了lambda表达式实现字典按键排序、按值排序、多条件排序相关操作技巧,需要的朋友可以参考下2019-07-07
不知道大家会不会遇到这样的情况，自己辛辛苦苦整理的攻略，分享给自己的一些朋友，结果分享有人堂而皇之地拿着这份攻略图片去引流，并声称是自己整理的，真是岂有此理！本文就来用Python实现为图片上版权，需要的可以参考一下2023-01-01
重写django的model下的objects模型管理器方式
这篇文章主要介绍了重写django的model下的objects模型管理器方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-05-05
最新评论