加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
get_mean_demo.py 729 Bytes
一键复制 编辑 原始数据 按行查看 历史
wangtonggui 提交于 2019-08-15 15:31 . 添加python获取均值的方法。
# coding: utf-8
'''
是用caffe的./build/tools/compute_image_mean -backend=$DBTYPE $EXAMPLE/cifar10_train_$DBTYPE $EXAMPLE/mean.binaryproto
可以获取均值文件。
那么均值到底长什么样子呢。
本脚本用python获取了均值,并输出其shape为:
(3, 32, 32)
表示了每个像素点位置的均值
'''
import numpy as np
import sys
import caffe
if len(sys.argv) != 3:
print "Usage: python convert_mean.py mean.binaryproto mean.npy"
sys.exit()
blob = caffe.proto.caffe_pb2.BlobProto()
bin_mean = open(sys.argv[1], 'rb').read()
blob.ParseFromString(bin_mean)
arr = np.array(caffe.io.blobproto_to_array(blob))
npy_mean = arr[0]
# (3, 32, 32)
print npy_mean.shape
# np.save(sys.argv[2], npy_mean)
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化