Caffe学习:使用pycaffe生成mean_file.py文件

来源:互联网 发布:数据库原理视频 编辑:程序博客网 时间:2024/05/17 06:11

利用build/tools/compute_image_mean产生mean.binaryproto文件后,利用如下代码产生mean.npy文件,给pycaffe调用:

#!/usr/bin/env pythonMEAN_BIN = 'mean.binaryproto'MEAN_NPY = 'mean.npy'from caffe.proto import caffe_pb2from caffe.io import blobproto_to_arrayprint('generating mean file...')mean_blob = caffe_pb2.BlobProto()mean_blob.ParseFromString(open(MEAN_BIN, 'rb').read())import numpy as npmean_npy = blobproto_to_array(mean_blob)mean_npy_shape = mean_npy.shapemean_npy = mean_npy.reshape(mean_npy_shape[1], mean_npy_shape[2], mean_npy_shape[3])np.save(file(MEAN_NPY, 'wb'), mean_npy)print('done...')
0 0
原创粉丝点击