gpt4 book ai didi

Caffe均值文件mean.binaryproto转mean.npy的方法

转载 作者:qq735679552 更新时间:2022-09-27 22:32:09 24 4
gpt4 key购买 nike

CFSDN坚持开源创造价值,我们致力于搭建一个资源共享平台,让每一个IT人在这里找到属于你的精彩世界.

这篇CFSDN的博客文章Caffe均值文件mean.binaryproto转mean.npy的方法由作者收集整理,如果你对这篇文章有兴趣,记得点赞哟.

mean.binaryproto转mean.npy 。

使用Caffe的C++接口进行操作时,需要的图像均值文件是pb格式,例如常见的均值文件名为mean.binaryproto;但在使用python接口进行操作时,需要的图像均值文件是numpy格式,例如mean.npy。所以在跨语言进行操作时,需要将mean.binaryproto转换成mean.npy,转换代码如下:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
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。代码如下:

?
1
2
3
4
5
6
7
8
9
10
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的代码如下:

?
1
2
3
4
import numpy as np
 
mean_npy = np.load(MEAN_NPY_PATH)
mean = mean_npy.mean( 1 ).mean( 1 )

以上这篇Caffe均值文件mean.binaryproto转mean.npy的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持我.

原文链接:https://blog.csdn.net/hyman_yx/article/details/51732656 。

最后此篇关于Caffe均值文件mean.binaryproto转mean.npy的方法的文章就讲到这里了,如果你想了解更多关于Caffe均值文件mean.binaryproto转mean.npy的方法的内容请搜索CFSDN的文章或继续浏览相关文章,希望大家以后支持我的博客! 。

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com