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

更新時間：2018年07月09日 09:04:14 作者：hyman_yx

今天小編就為大家分享一篇Caffe均值文件mean.binaryproto轉(zhuǎn)mean.npy的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

mean.binaryproto轉(zhuǎn)mean.npy

使用Caffe的C++接口進行操作時，需要的圖像均值文件是pb格式，例如常見的均值文件名為mean.binaryproto；但在使用python接口進行操作時，需要的圖像均值文件是numpy格式，例如mean.npy。所以在跨語言進行操作時，需要將mean.binaryproto轉(zhuǎn)換成mean.npy，轉(zhuǎn)換代碼如下：

import caffe
import numpy as np

MEAN_PROTO_PATH = 'mean.binaryproto'        # 待轉(zhuǎn)換的pb格式圖像均值文件路徑
MEAN_NPY_PATH = 'mean.npy'             # 轉(zhuǎn)換后的numpy格式圖像均值文件路徑

blob = caffe.proto.caffe_pb2.BlobProto()      # 創(chuàng)建protobuf blob
data = open(MEAN_PROTO_PATH, 'rb' ).read()     # 讀入mean.binaryproto文件內(nèi)容
blob.ParseFromString(data)             # 解析文件內(nèi)容到blob

array = np.array(caffe.io.blobproto_to_array(blob))# 將blob中的均值轉(zhuǎn)換成numpy格式，array的shape （mean_number，channel, hight, width）
mean_npy = array[0]                # 一個array中可以有多組均值存在，故需要通過下標選擇其中一組均值
np.save(MEAN_NPY_PATH ,mean_npy)

已知圖像均值，構(gòu)造mean.npy

如果已知圖像中每個通道的均值，例如3通道圖像每個通道的均值分別為104，117，123，我們也可以通過其構(gòu)造mean.npy。代碼如下：

import numpy as np

MEAN_NPY_PATH = 'mean.npy'

mean = np.ones([3,256, 256], dtype=np.float)
mean[0,:,:] = 104
mean[1,:,:] = 117
mean[2,:,:] = 123

np.save(MEAN_NPY, mean)

載入mean.npy

上面我們用兩種方式構(gòu)造了均值文件mean.npy，在使用時載入mean.npy的代碼如下：

import numpy as np

mean_npy = np.load(MEAN_NPY_PATH)
mean = mean_npy.mean(1).mean(1)

以上這篇Caffe均值文件mean.binaryproto轉(zhuǎn)mean.npy的方法就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

相關(guān)文章

python實現(xiàn)矩陣乘法的方法
這篇文章主要介紹了python實現(xiàn)矩陣乘法的方法,實例分析了Python矩陣操作的相關(guān)技巧,需要的朋友可以參考下
2015-06-06
numpy中nan_to_num的具體使用
在Numpy中NaN值一般出現(xiàn)在數(shù)據(jù)清洗前，出現(xiàn)這個值說明這個數(shù)據(jù)是缺失的，本文主要介紹了numpy中nan_to_num的具體使用，感興趣的可以了解一下
2022-08-08
python 根據(jù)字典的鍵值進行排序的方法
這篇文章主要介紹了python 根據(jù)字典的鍵值進行排序的實現(xiàn)方法，本文給大家介紹的非常詳細，具有一定的參考借鑒價值 ,需要的朋友可以參考下
2019-07-07
python使用循環(huán)打印所有三位數(shù)水仙花數(shù)的實例
今天小編就為大家分享一篇python使用循環(huán)打印所有三位數(shù)水仙花數(shù)的實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2018-11-11
Pyinstaller打包多個資源文件的超詳細教程(推薦!)
Pyinstaller是著名python打包module,軟件或者小工具用python寫好后,用pyinstaller 可以實現(xiàn)輕松打包,下面這篇文章主要給大家介紹了關(guān)于Pyinstaller打包多個資源文件的超詳細教程,需要的朋友可以參考下
2022-12-12
Python離線安裝openpyxl模塊的步驟
這篇文章主要介紹了Python離線安裝openpyxl模塊的步驟,本文給大家介紹的非常詳細，對大家的學(xué)習或工作具有一定的參考借鑒價值，需要的朋友可以參考下
2021-03-03
Python 面向?qū)ο笾恈lass和對象基本用法示例
這篇文章主要介紹了Python 面向?qū)ο笾恈lass和對象基本用法,結(jié)合實例形式詳細分析了Python面向?qū)ο蟪绦蛟O(shè)計中類class和對象基本概念、原理、使用方法與操作注意事項,需要的朋友可以參考下
2020-02-02
python實現(xiàn)簡單聊天室功能可以私聊
這篇文章主要為大家詳細介紹了python實現(xiàn)簡單聊天室功能，可以進行私聊，具有一定的參考價值，感興趣的小伙伴們可以參考一下
2019-07-07
Python之標點符號string.punctuation的使用
Python的string模塊提供了一個方便的屬性string.punctuation,其中包含所有ASCII標點符號字符,這使得在處理和識別字符串中的標點符號時非常有用,可以通過簡單的in關(guān)鍵字來檢測字符是否為標點
2024-09-09
分享python數(shù)據(jù)統(tǒng)計的一些小技巧
今天這些小技巧在處理python的一些數(shù)據(jù)方面還是很有幫助的，希望能幫到在這方面有需要的童鞋~
2016-07-07