caffe binaryproto 與 npy相互轉(zhuǎn)換的實例講解

更新時間：2018年07月09日 09:12:59 作者：無眠梔

今天小編就為大家分享一篇caffe binaryproto 與 npy相互轉(zhuǎn)換的實例講解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

在caffe中，如果使用的是c++接口，均值文件默認(rèn)為.binaryproto格式，而如果使用的是python接口，均值文件默認(rèn)的是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)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

Caffe均值文件mean.binaryproto轉(zhuǎn)mean.npy的方法

相關(guān)文章

pandas中Timestamp類用法詳解
這篇文章主要為大家詳細(xì)介紹了pandas中Timestamp類用法，具有一定的參考價值，感興趣的小伙伴們可以參考一下
2017-12-12
Python3 使用pillow庫生成隨機驗證碼
這篇文章主要介紹了Python3 使用pillow庫生成隨機驗證碼,需要的朋友可以參考下
2019-08-08
TensorFlow實現(xiàn)卷積神經(jīng)網(wǎng)絡(luò)CNN
這篇文章主要為大家詳細(xì)介紹了TensorFlow實現(xiàn)卷積神經(jīng)網(wǎng)絡(luò)CNN，具有一定的參考價值，感興趣的小伙伴們可以參考一下
2018-03-03
Python 異步之生成器示例詳解
這篇文章主要為大家介紹了Python 異步之生成器示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪
2023-03-03
python使用js2py庫運行js代碼
本文主要介紹了thon使用js2py庫運行js代碼，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
2022-05-05
Django2.1集成xadmin管理后臺所遇到的錯誤集錦(填坑)
這篇文章主要介紹了Django2.1集成xadmin管理后臺所遇到的錯誤集錦(填坑),小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧
2018-12-12
django美化后臺django-suit的安裝配置操作
這篇文章主要介紹了django美化后臺django-suit的安裝配置操作，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2020-07-07
Python使用lambda表達式對字典排序操作示例
這篇文章主要介紹了Python使用lambda表達式對字典排序操作,結(jié)合實例形式分析了lambda表達式實現(xiàn)字典按鍵排序、按值排序、多條件排序相關(guān)操作技巧,需要的朋友可以參考下
2019-07-07
一行Python代碼實現(xiàn)為圖片上版權(quán)
不知道大家會不會遇到這樣的情況，自己辛辛苦苦整理的攻略，分享給自己的一些朋友，結(jié)果分享有人堂而皇之地拿著這份攻略圖片去引流，并聲稱是自己整理的，真是豈有此理！本文就來用Python實現(xiàn)為圖片上版權(quán)，需要的可以參考一下
2023-01-01
重寫django的model下的objects模型管理器方式
這篇文章主要介紹了重寫django的model下的objects模型管理器方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2020-05-05