如何将.npy文件转换为.binaryproto？

Question

我使用python创建了一个平均图像文件并将其保存到numpy文件中。我想知道如何将此.npy文件转换为.binaryproto文件。我正在使用此文件来使用GoogLeNet进行培训。

Answer 1

你可以简单地使用numpy来创建.binaryproto和给定的caffe io函数

import caffe
#avg_img is your numpy array with the average data 
blob = caffe.io.array_to_blobproto( avg_img)
with open( mean.binaryproto, 'wb' ) as f :
    f.write( blob.SerializeToString())

Answer 2

这是@ Kev1n91代码的改进版。

import caffe
import numpy as np

mean_npy = np.load('mean.npy') # Input numpy array
blob = caffe.io.array_to_blobproto(mean_npy)
mean_binproto = 'mean.binaryproto' # Output binaryproto file
with open(mean_binproto, 'wb') as f :
    f.write( blob.SerializeToString())