caffe binaryproto 与 npy相互转换的实例讲解


Posted in Python onJuly 09, 2018

在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相互转换的实例讲解就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
在Linux中通过Python脚本访问mdb数据库的方法
May 06 Python
详细解析Python中的变量的数据类型
May 13 Python
在Python中使用swapCase()方法转换大小写的教程
May 20 Python
python通过ssh-powershell监控windows的方法
Jun 02 Python
Python绘制的二项分布概率图示例
Aug 22 Python
Python3.6中Twisted模块安装的问题与解决
Apr 15 Python
python Pandas如何对数据集随机抽样
Jul 29 Python
python模块导入的方法
Oct 24 Python
pytorch:实现简单的GAN示例(MNIST数据集)
Jan 10 Python
pytorch进行上采样的种类实例
Feb 18 Python
Pandas中DataFrame基本函数整理(小结)
Jul 20 Python
Python实现给PDF添加水印的方法
Jan 25 Python
利用Python如何生成便签图片详解
Jul 09 #Python
Caffe均值文件mean.binaryproto转mean.npy的方法
Jul 09 #Python
使用Python更换外网IP的方法
Jul 09 #Python
使用Python AIML搭建聊天机器人的方法示例
Jul 09 #Python
Python迭代器与生成器用法实例分析
Jul 09 #Python
numpy的文件存储.npy .npz 文件详解
Jul 09 #Python
Python实现聊天机器人的示例代码
Jul 09 #Python
You might like
PHP图片等比例缩放生成缩略图函数分享
2014/06/10 PHP
php实现粘贴截图并完成上传功能
2015/05/17 PHP
Laravel路由设定和子路由设定实例分析
2016/03/30 PHP
windows环境下使用Composer安装ThinkPHP5
2018/05/18 PHP
javascript数组去掉重复
2011/05/12 Javascript
AngularJS语法详解(续)
2015/01/23 Javascript
JavaScript实现数据类型的相互转换
2016/03/06 Javascript
判断是否存在子节点的实现代码
2016/05/18 Javascript
jQuery自适应轮播图插件Swiper用法示例
2016/08/24 Javascript
基于JS实现的随机数字抽签实例
2016/12/08 Javascript
vue移动端弹框组件的实例
2018/09/25 Javascript
[01:00:22]DOTA2-DPC中国联赛定级赛 LBZS vs Magma BO3第三场 1月10日
2021/03/11 DOTA
Python的Flask框架应用调用Redis队列数据的方法
2016/06/06 Python
python魔法方法-属性访问控制详解
2016/07/25 Python
python的mysqldb安装步骤详解
2017/08/14 Python
详解python基础之while循环及if判断
2017/08/24 Python
分享一个简单的python读写文件脚本
2017/11/25 Python
Python设计模式之命令模式简单示例
2018/01/10 Python
python实现连续图文识别
2018/12/18 Python
Python利用lxml模块爬取豆瓣读书排行榜的方法与分析
2019/04/15 Python
django query模块
2019/04/20 Python
python2.7 安装pip的方法步骤(管用)
2019/05/05 Python
Python Pandas中根据列的值选取多行数据
2019/07/08 Python
详解Django CAS 解决方案
2019/10/30 Python
PyCharm Ctrl+Shift+F 失灵的简单有效解决操作
2021/01/15 Python
使用pandas读取表格数据并进行单行数据拼接的详细教程
2021/03/03 Python
突袭HTML5之Javascript API扩展1—Web Worker异步执行及相关概述
2013/01/31 HTML / CSS
Steve Madden官网:美国鞋类品牌
2017/01/29 全球购物
Origins悦木之源英国官网:雅诗兰黛集团高端植物护肤品牌
2017/11/06 全球购物
学生励志演讲稿
2014/01/06 职场文书
小学后勤管理制度
2014/01/14 职场文书
机械设计毕业生自荐信
2014/02/02 职场文书
申论倡议书范文
2014/05/13 职场文书
学校安全防火方案
2014/06/07 职场文书
Mysql分库分表之后主键处理的几种方法
2022/02/15 MySQL
Python 第三方库 openpyxl 的安装过程
2022/12/24 Python