Как преобразовать .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())