欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

python與caffe改變通道順序的方法

 更新時(shí)間:2018年08月04日 11:57:30   作者:ShellCollector  
今天小編就為大家分享一篇python與caffe改變通道順序的方法,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧

把通道放在前面:

image = cv2.imread(path + file)

image = cv2.resize(image, (48, 48), interpolation=cv2.INTER_CUBIC) aaaa= np.transpose(image,(2, 0, 1)) print(aaaa)

圖像原來shape:(48, 48, 3),改之后shape:(3,48,48)

注意:reshape不能解決通道轉(zhuǎn)換問題

pycaffe做識(shí)別時(shí)通道轉(zhuǎn)換問題:

要注意一點(diǎn)的就是:Caffe中彩色圖像的通道是BGR格式,圖像存儲(chǔ)是【0,255】

1.caffe.io.load_image方式 view plai cop

image = caffe.io.load_image(image_file) #加載圖片

使用caffe.io.load_image()讀進(jìn)來的是RGB格式和0~1(float)

所以在進(jìn)行識(shí)別之前要在transformer中設(shè)置transformer.set_raw_scale('data',255)(縮放至0~255)

以及transformer.set_channel_swap('data',(2,1,0)(將RGB變換到BGR)

# python中將圖片存儲(chǔ)為[0, 1],而caffe中將圖片存儲(chǔ)為[0, 255],所以需要一個(gè)轉(zhuǎn)換 
transformer.set_raw_scale('data', 255)  # 縮放到[0,255]之間 
transformer.set_channel_swap('data', (2,1,0)) #交換通道,將圖片由RGB變?yōu)锽GR(caffe中圖片是BGR格式,而原始格式是RGB,所以要轉(zhuǎn)化) 

2.使用cv2.imread()來讀取圖片

cv2.imread()接口讀圖像,讀進(jìn)來直接是BGR 格式and 0~255

所以不需要再縮放到【0,255】和通道變換【2,1,0】,不需要transformer.set_raw_scale('data',255)和transformer.set_channel_swap('data',(2,1,0))

3.使用PIL來讀取圖片

對(duì)于彩色圖像,不管其圖像格式是PNG,還是BMP,或者JPG,在PIL中,使用Image模塊的open()函數(shù)打開后,返回的圖像對(duì)象的模式都是“RGB”。而對(duì)于灰度圖像,不管其圖像格式是PNG,還是BMP,或者JPG,打開后,其模式為“L”。所以需要轉(zhuǎn)換格式,但不需要縮放到[0,255]

data = np.array(Image.open(self.dataRoot+img_list)) 
data = np.transpose(data,(2,0,1))#轉(zhuǎn)換通道 
data[[0,2],...] = data[[2,0],...] #RGB→BGR 

4.對(duì)于matlab來說

Caffe中的blobs格式是N*C*H*W,分別是數(shù)量Number,通道數(shù)Channel,以及寬度Height和寬度Width

而matlab中是先寬后高,即[w,h],圖像的通道是RGB

所以需要進(jìn)行相應(yīng)的轉(zhuǎn)換:

im_data = im (:,:,[3,2,1]) ; %RGB to BGR

im_data = permute(im_data, [2,1,3]); %旋轉(zhuǎn)高度和寬度

最后,分享一個(gè)Caffe的典型python識(shí)別代碼:

# -*- coding: utf-8 -*- 
""" 
Created on Sun May 28 16:00:47 2017 
@author: fancp,#windows下CPU模式 
""" 
import numpy as np 
import caffe 
import sys 
caffe_root = 'F:/Caffe' #########你自己的Caffe路徑 
sys.path.insert(0, caffe_root + '/python') 
 
size = 227 #訓(xùn)練的圖片尺寸 
image_file = 'F:/.../.../nihao.jpg'#圖片路徑 
model_def = 'F:/.../.../deploy.prototxt'#deploy模型文件位置 
model_weights = 'F:/.../.../_iter_20000.caffemodel'#訓(xùn)練完的模型位置 
net = caffe.Net(model_def, model_weights, caffe.TEST)  
 
# 加載均值文件 
mu = np.load(caffe_root + '/python/caffe/imagenet/ilsvrc_2012_mean.npy') ###caffe 自帶的文件 
mu = mu.mean(1).mean(1) # average over pixels to obtain the mean (BGR) pixel values 
###########################下面這5句等同與上面兩句,選擇其一################# 
#blob = caffe.proto.caffe_pb2.BlobProto() 
#mean_data = open( 'mean.binaryproto' , 'rb' ).read() 
#blob.ParseFromString(mean_data) 
#mu = np.array(caffe.io.blobproto_to_array(blob)) 
#mu = mu.mean(1).mean(1).mean(1) 
############################################################################## 
#圖片預(yù)處理 
transformer = caffe.io.Transformer({'data': net.blobs['data'].data.shape}) ##設(shè)定圖片的shape格式(1,3,227,227),大小由deploy 文件指定 
# python讀取的圖片文件格式為H×W×K,需轉(zhuǎn)化為K×H×W 
transformer.set_transpose('data', (2,0,1)) #改變維度的順序,由原始圖片(227,227,3)變?yōu)?3,227,227) 
transformer.set_mean('data', mu)   # 每個(gè)通道減去均值 
 
# python中將圖片存儲(chǔ)為[0, 1],而caffe中將圖片存儲(chǔ)為[0, 255],所以需要一個(gè)轉(zhuǎn)換 
transformer.set_raw_scale('data', 255)  # 縮放到【0,255】之間 
transformer.set_channel_swap('data', (2,1,0)) #交換通道,將圖片由RGB變?yōu)锽GR(caffe中圖片是BGR格式,而原始格式是RGB,所以要轉(zhuǎn)化) 
net.blobs['data'].reshape(1,3,size, size) # 將輸入圖片格式轉(zhuǎn)化為合適格式(與deploy文件相同) 
#上面這句,第一參數(shù):圖片數(shù)量 第二個(gè)參數(shù) :通道數(shù) 第三個(gè)參數(shù):圖片高度 第四個(gè)參數(shù):圖片寬度 
 
image = caffe.io.load_image(image_file) #加載圖片 
# 用上面的transformer.preprocess來處理剛剛加載圖片 
net.blobs['data'].data[...] = transformer.preprocess('data', image)  
 
### perform classification 
caffe.set_mode_cpu() 
output = net.forward() 
#print output 
output_prob = output['prob'][0].argmax() # 給出概率最高的是第幾類,需要自己對(duì)應(yīng)到我們約定的類別去 

以上這篇python與caffe改變通道順序的方法就是小編分享給大家的全部?jī)?nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。

相關(guān)文章

  • python 實(shí)現(xiàn)12bit灰度圖像映射到8bit顯示的方法

    python 實(shí)現(xiàn)12bit灰度圖像映射到8bit顯示的方法

    這篇文章主要介紹了python 實(shí)現(xiàn)12bit灰度圖像映射到8bit顯示的方法,文中通過示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
    2019-07-07
  • python自動(dòng)化調(diào)用百度api解決驗(yàn)證碼

    python自動(dòng)化調(diào)用百度api解決驗(yàn)證碼

    這篇文章主要介紹了python自動(dòng)化調(diào)用百度api解決驗(yàn)證碼,文中通過示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
    2021-04-04
  • 一起來學(xué)習(xí)一下python的數(shù)據(jù)類型

    一起來學(xué)習(xí)一下python的數(shù)據(jù)類型

    這篇文章主要為大家詳細(xì)介紹了python的數(shù)據(jù)類型,文中示例代碼介紹的非常詳細(xì),具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下希望能夠給你帶來幫助
    2022-01-01
  • python 實(shí)現(xiàn)邏輯回歸

    python 實(shí)現(xiàn)邏輯回歸

    這篇文章主要介紹了python 實(shí)現(xiàn)邏輯回歸的方法,幫助大家更好的理解和使用python,感興趣的朋友可以了解下
    2020-12-12
  • Python如何讀寫二進(jìn)制數(shù)組數(shù)據(jù)

    Python如何讀寫二進(jìn)制數(shù)組數(shù)據(jù)

    這篇文章主要介紹了Python如何讀寫二進(jìn)制數(shù)組數(shù)據(jù),文中講解非常細(xì)致,代碼幫助大家更好的理解和學(xué)習(xí),感興趣的朋友可以了解下
    2020-08-08
  • python pygame實(shí)現(xiàn)方向鍵控制小球

    python pygame實(shí)現(xiàn)方向鍵控制小球

    這篇文章主要為大家詳細(xì)介紹了python pygame實(shí)現(xiàn)方向鍵控制小球,具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下
    2019-05-05
  • 簡(jiǎn)析Python函數(shù)式編程字符串和元組及函數(shù)分類與高階函數(shù)

    簡(jiǎn)析Python函數(shù)式編程字符串和元組及函數(shù)分類與高階函數(shù)

    這篇文章主要介紹了Python函數(shù)式編程中的字符串、元組及函數(shù)分類與高階函數(shù),有需要的朋友可以借鑒參考下,希望能夠有所幫助
    2021-09-09
  • Python獲取本機(jī)IP/MAC多網(wǎng)卡方法示例

    Python獲取本機(jī)IP/MAC多網(wǎng)卡方法示例

    這篇文章主要為大家介紹了Python獲取本機(jī)IP/MAC多網(wǎng)卡方法示例,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪
    2023-08-08
  • Python中pandas的dataframe過濾數(shù)據(jù)方法

    Python中pandas的dataframe過濾數(shù)據(jù)方法

    這篇文章主要介紹了Python中pandas的dataframe過濾數(shù)據(jù)方法,Pandas是另外一個(gè)用于處理高級(jí)數(shù)據(jù)結(jié)構(gòu)和數(shù)據(jù)分析的Python庫(kù),Pandas是基于Numpy構(gòu)建的一種工具,需要的朋友可以參考下
    2023-07-07
  • Python類的定義和使用詳情

    Python類的定義和使用詳情

    這篇文章主要介紹了Python類的定義與使用,類名只要是一個(gè)合法的標(biāo)識(shí)符即可,但這僅僅滿足的是?Python?的語(yǔ)法要求:如果從程序的可讀性方面來看,Python?的類名必須是由一個(gè)或多個(gè)有意義的單詞連綴而成的,下文基于這些基礎(chǔ)內(nèi)容展開介紹,需要的朋友可以參考一下
    2022-03-03

最新評(píng)論