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的方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

教你两步解决conda安装pytorch时下载速度慢or超时的问题
这篇文章主要介绍了教你两步解决conda安装pytorch时下载速度慢or超时的问题,使用清华镜像源可以大大减少安装的时间,需要的朋友可以参考下
2023-03-03
Pytorch四维Tensor转图片并保存方式(维度顺序调整)
这篇文章主要介绍了Pytorch四维Tensor转图片并保存方式(维度顺序调整)，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
2022-12-12
python3.6的venv模块使用详解
这篇文章主要介绍了python3.6的venv模块使用详解，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2018-08-08
Pyqt5打开电脑摄像头进行拍照的实现示例
本文介绍了如何使用Pyqt5来控制摄像头拍照，通过构建一个简单的用户界面，我们可以实现从摄像头实时获取图像，保存图片，感兴趣的可以了解一下
2023-08-08
超详细注释之OpenCV制作图像Mask
这篇文章主要介绍了OpenCV制作图像Mask，本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
2021-09-09
Python爬虫之Selenium鼠标事件的实现
这篇文章主要介绍了Python爬虫之Selenium鼠标事件的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-12-12
python代码如何调出cmd命令行窗口并在cmd窗口中执行指定的命令
这篇文章主要介绍了python代码如何调出cmd命令行窗口并在cmd窗口中执行指定的命令问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
2024-02-02
PySpark和RDD对象最新详解
Spark是一款分布式的计算框架，用于调度成百上千的服务器集群，计算TB、PB乃至EB级别的海量数据，PySpark是由Spark官方开发的Python语言第三方库，本文重点介绍PySpark和RDD对象，感兴趣的朋友一起看看吧
2023-01-01
Python用sndhdr模块识别音频格式详解
这篇文章主要介绍了Python用sndhdr模块识别音频格式详解，具有一定借鉴价值,需要的朋友可以参考下
2018-01-01
使用Python实现tail的示例代码
tail是一个常用的Linux命令, 它可以打印文件的后面n行数据, 也能实时输出文件的追加数据。本文就来用Python实现tail，感兴趣的可以了解一下
2023-03-03