python與caffe改變通道順序的方法
把通道放在前面:
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顯示的方法,文中通過示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-07-07python自動(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ù)類型
這篇文章主要為大家詳細(xì)介紹了python的數(shù)據(jù)類型,文中示例代碼介紹的非常詳細(xì),具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下希望能夠給你帶來幫助2022-01-01Python如何讀寫二進(jìn)制數(shù)組數(shù)據(jù)
這篇文章主要介紹了Python如何讀寫二進(jìn)制數(shù)組數(shù)據(jù),文中講解非常細(xì)致,代碼幫助大家更好的理解和學(xué)習(xí),感興趣的朋友可以了解下2020-08-08python pygame實(shí)現(xiàn)方向鍵控制小球
這篇文章主要為大家詳細(xì)介紹了python pygame實(shí)現(xiàn)方向鍵控制小球,具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下2019-05-05簡(jiǎn)析Python函數(shù)式編程字符串和元組及函數(shù)分類與高階函數(shù)
這篇文章主要介紹了Python函數(shù)式編程中的字符串、元組及函數(shù)分類與高階函數(shù),有需要的朋友可以借鑒參考下,希望能夠有所幫助2021-09-09Python獲取本機(jī)IP/MAC多網(wǎng)卡方法示例
這篇文章主要為大家介紹了Python獲取本機(jī)IP/MAC多網(wǎng)卡方法示例,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2023-08-08Python中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