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使用PyQt5/PySide2编写一个极简的音乐播放器功能
这篇文章主要介绍了Python中使用PyQt5/PySide2编写一个极简的音乐播放器功能，本文通过实例代码给大家介绍的非常详细，具有一定的参考借鉴价值,需要的朋友可以参考下2020-02-02
这篇文章主要介绍了Python实现功能完整的个人员管理程序，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习吧2022-12-12
在本篇文章中小编给大家整理的是一篇关于go语言和python的区别点，需要的朋友们可以学习下。2020-07-07
在数据分析时,经常需要将数据分成不同的群组,pandas中的groupby()函数可以完美地完成各种分组操作,本文就来介绍一下Pandas分组函数groupby的用法,感兴趣的可以了解一下2024-01-01
python中自定义异常/raise关键字抛出异常的案例解析
在编程过程中合理的使用异常可以使得程序正常的执行，本篇文章给大家介绍python中自定义异常/raise关键字抛出异常案例解析,需要的朋友可以参考下2024-01-01
这篇文章主要介绍了Python执行流程控制，流程控制即控制流程，具体指控制程序的执行流程，而程序的执行流程分为三种结构：顺序结构、分支结构、循环结构，下文详细介绍需要的小伙伴可以参考一下2022-04-04
这篇文章主要介绍了pyinstaller将python程序打包为可执行文件，pyinstaller是一个python打包工具，它将python程序及所需依赖都打包成一个可执行文件2022-08-08
这篇文章主要为大家详细介绍了pygame实现烟雨蒙蒙下彩虹雨，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-11-11
这篇文章主要为大家详细介绍了Python实现单向链表，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2022-05-05
这篇文章主要介绍了python绘制云雨图raincloud plot，Raincloud的Python实现是一个名为PtitPrince的包，它写在seaborn之上，这是一个Python绘图库，用于从pandas数据帧中获取漂亮的绘图2022-08-08
最新评论