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 相关文章推荐
python使用beautifulsoup从爱奇艺网抓取视频播放
Jan 23 Python
Python中的rfind()方法使用详解
May 19 Python
浅析Python基础-流程控制
Mar 18 Python
Python 爬虫模拟登陆知乎
Sep 23 Python
Python实现合并同一个文件夹下所有txt文件的方法示例
Apr 26 Python
python 读取文件并替换字段的实例
Jul 12 Python
在Python中pandas.DataFrame重置索引名称的实例
Nov 06 Python
python GUI库图形界面开发之PyQt5菜单栏控件QMenuBar的详细使用方法与实例
Feb 28 Python
python中setuptools的作用是什么
Jun 19 Python
Python 如何创建一个简单的REST接口
Jul 30 Python
学习Python爬虫的几点建议
Aug 05 Python
python mock测试的示例
Oct 19 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实现基于微信公众平台开发SDK(demo)扩展的方法
2014/12/22 PHP
注意!PHP 7中不要做的10件事
2016/09/18 PHP
php闭包中使用use声明变量的作用域实例分析
2018/08/09 PHP
CLASS_CONFUSION JS混淆 全源码
2007/12/12 Javascript
express的中间件cookieParser详解
2014/12/04 Javascript
JS获取文件大小方法小结
2015/12/08 Javascript
js简单网速测试方法完整实例
2015/12/15 Javascript
JavaScript实现简单的tab选项卡切换
2016/01/05 Javascript
深入学习jQuery Validate表单验证(二)
2016/01/18 Javascript
jquery日历插件datepicker用法分析
2016/01/22 Javascript
jQuery旋转插件jqueryrotate用法详解
2016/10/13 Javascript
使用Node.js给图片加水印的方法
2016/11/15 Javascript
原生js实现对Ajax的封装(仿jquery)
2017/01/22 Javascript
canvas实现钟表效果
2017/02/13 Javascript
javascript设计模式之策略模式学习笔记
2017/02/15 Javascript
nodejs 搭建简易服务器的图文教程(推荐)
2017/07/18 NodeJs
微信jssdk逻辑在vue中的运用详解
2018/11/14 Javascript
原来JS还可以这样拆箱转换详解
2019/02/01 Javascript
python网络编程学习笔记(四):域名系统
2014/06/09 Python
python连接数据库的方法
2017/10/19 Python
python操作列表的函数使用代码详解
2017/12/28 Python
Django框架HttpRequest对象用法实例分析
2019/11/01 Python
利用CSS3的flexbox实现水平垂直居中与三列等高布局
2016/09/12 HTML / CSS
香港No.1得奖零食网:香港零食大王
2016/07/22 全球购物
纽约复古灵感的现代珠宝品牌:Lulu Frost
2018/03/03 全球购物
美国林业供应商:Forestry Suppliers
2019/05/01 全球购物
linux面试题参考答案(9)
2016/01/29 面试题
幼儿园校车司机的岗位职责
2014/01/30 职场文书
艺术设计专业求职自荐信
2014/05/19 职场文书
护理实习生带教计划
2015/01/16 职场文书
销售经理岗位职责范本
2015/04/02 职场文书
2015年乡镇工作总结范文
2015/04/22 职场文书
爱国电影观后感
2015/06/19 职场文书
新闻稿件写作技巧
2015/07/18 职场文书
导游词之湖州-太湖
2019/10/11 职场文书
Go Grpc Gateway兼容HTTP协议文档自动生成网关
2022/06/16 Golang