Caffe均值文件mean.binaryproto转mean.npy的方法
更新时间：2018年07月09日 09:04:14 作者：hyman_yx
今天小编就为大家分享一篇Caffe均值文件mean.binaryproto转mean.npy的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
mean.binaryproto转mean.npy
使用Caffe的C++接口进行操作时，需要的图像均值文件是pb格式，例如常见的均值文件名为mean.binaryproto；但在使用python接口进行操作时，需要的图像均值文件是numpy格式，例如mean.npy。所以在跨语言进行操作时，需要将mean.binaryproto转换成mean.npy，转换代码如下：
import caffe import numpy as np MEAN_PROTO_PATH = 'mean.binaryproto' # 待转换的pb格式图像均值文件路径 MEAN_NPY_PATH = 'mean.npy' # 转换后的numpy格式图像均值文件路径 blob = caffe.proto.caffe_pb2.BlobProto() # 创建protobuf blob data = open(MEAN_PROTO_PATH, 'rb' ).read() # 读入mean.binaryproto文件内容 blob.ParseFromString(data) # 解析文件内容到blob array = np.array(caffe.io.blobproto_to_array(blob))# 将blob中的均值转换成numpy格式，array的shape （mean_number，channel, hight, width） mean_npy = array[0] # 一个array中可以有多组均值存在，故需要通过下标选择其中一组均值 np.save(MEAN_NPY_PATH ,mean_npy)
已知图像均值，构造mean.npy
如果已知图像中每个通道的均值，例如3通道图像每个通道的均值分别为104，117，123，我们也可以通过其构造mean.npy。代码如下：
import numpy as np MEAN_NPY_PATH = 'mean.npy' mean = np.ones([3,256, 256], dtype=np.float) mean[0,:,:] = 104 mean[1,:,:] = 117 mean[2,:,:] = 123 np.save(MEAN_NPY, mean)
载入mean.npy
上面我们用两种方式构造了均值文件mean.npy，在使用时载入mean.npy的代码如下：
import numpy as np mean_npy = np.load(MEAN_NPY_PATH) mean = mean_npy.mean(1).mean(1)
以上这篇Caffe均值文件mean.binaryproto转mean.npy的方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了python实现矩阵乘法的方法,实例分析了Python矩阵操作的相关技巧,需要的朋友可以参考下2015-06-06
在Numpy中NaN值一般出现在数据清洗前，出现这个值说明这个数据是缺失的，本文主要介绍了numpy中nan_to_num的具体使用，感兴趣的可以了解一下2022-08-08
这篇文章主要介绍了python 根据字典的键值进行排序的实现方法，本文给大家介绍的非常详细，具有一定的参考借鉴价值 ,需要的朋友可以参考下2019-07-07
今天小编就为大家分享一篇python使用循环打印所有三位数水仙花数的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-11-11
Pyinstaller打包多个资源文件的超详细教程(推荐!)
Pyinstaller是著名python打包module,软件或者小工具用python写好后,用pyinstaller 可以实现轻松打包,下面这篇文章主要给大家介绍了关于Pyinstaller打包多个资源文件的超详细教程,需要的朋友可以参考下2022-12-12
这篇文章主要介绍了Python离线安装openpyxl模块的步骤,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2021-03-03
这篇文章主要介绍了Python 面向对象之类class和对象基本用法,结合实例形式详细分析了Python面向对象程序设计中类class和对象基本概念、原理、使用方法与操作注意事项,需要的朋友可以参考下2020-02-02
这篇文章主要为大家详细介绍了python实现简单聊天室功能，可以进行私聊，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-07-07
Python之标点符号string.punctuation的使用
Python的string模块提供了一个方便的属性string.punctuation,其中包含所有ASCII标点符号字符,这使得在处理和识别字符串中的标点符号时非常有用,可以通过简单的in关键字来检测字符是否为标点2024-09-09
今天这些小技巧在处理python的一些数据方面还是很有帮助的，希望能帮到在这方面有需要的童鞋~2016-07-07
最新评论