python-如何将.npy文件转换为.binaryproto?

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

解决方法:

您可以简单地使用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())
上一篇:(转)深度学习入门路线


下一篇:编译caffe时出错,add_executable cannot create target “prior_box_layer“