caffe binaryproto 與 npy相互轉(zhuǎn)換的實例講解
在caffe中,如果使用的是c++接口,均值文件默認為.binaryproto格式,而如果使用的是python接口,均值文件默認的是numpy的.npy格式,在工作中有時需要將兩者進行互相轉(zhuǎn)換,具體方式如下:
binaryproto -> npy
import numpy as np import caffe import sys blob = caffe.proto.caffe_pb2.BlobProto() data = open( 'mean.binaryproto' , 'rb' ).read() blob.ParseFromString(data) arr = np.array( caffe.io.blobproto_to_array(blob) ) out = arr[0] np.save( 'mean.npy' , out )
npy -> binaryproto
data_mean is 1*H*W
import numpy as np import caffe import sys blob = caffe.proto.caffe_pb2.BlobProto() with open('mean.npy','rb') as f: data_mean = numpy.load(f) blob.channels=1 blob.height = data_mean.shape[0] blob.width = data_mean.shape[1] blob.data.extend(data_mean.astype(float).flat) binaryproto_file = open('mean.binaryproto', 'wb' ) binaryproto_file.write(blob.SerializeToString()) binaryproto_file.close()
以上這篇caffe binaryproto 與 npy相互轉(zhuǎn)換的實例講解就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關(guān)文章
pandas按條件篩選數(shù)據(jù)的實現(xiàn)
這篇文章主要介紹了pandas按條件篩選數(shù)據(jù)的實現(xiàn),文中通過示例代碼介紹的非常詳細,對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2021-02-02python實現(xiàn)word文檔批量轉(zhuǎn)成自定義格式的excel文檔的思路及實例代碼
這篇文章主要介紹了python實現(xiàn)word文檔批量轉(zhuǎn)成自定義格式的excel文檔的解決思路及實例代碼,代碼簡單易懂,非常不錯,具有一定的參考借鑒價值,需要的朋友可以參考下2020-02-02Python matplotlib繪圖可視化知識點整理(小結(jié))
這篇文章主要介紹了Python matplotlib繪圖可視化知識點整理(小結(jié)),小編覺得挺不錯的,現(xiàn)在分享給大家,也給大家做個參考。一起跟隨小編過來看看吧2018-03-03