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

How to convert .npy file into .binaryproto?

本文关键字：转换 binaryproto 文件 npy      更新时间：2023-10-16

我使用 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())

这是@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())
相关文章：