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-07
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ù)類型
這篇文章主要為大家詳細(xì)介紹了python的數(shù)據(jù)類型,文中示例代碼介紹的非常詳細(xì),具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下希望能夠給你帶來幫助2022-01-01
Python如何讀寫二進(jìn)制數(shù)組數(shù)據(jù)
這篇文章主要介紹了Python如何讀寫二進(jìn)制數(shù)組數(shù)據(jù),文中講解非常細(xì)致,代碼幫助大家更好的理解和學(xué)習(xí),感興趣的朋友可以了解下2020-08-08
python 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-09
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ù)方法,Pandas是另外一個(gè)用于處理高級(jí)數(shù)據(jù)結(jié)構(gòu)和數(shù)據(jù)分析的Python庫,Pandas是基于Numpy構(gòu)建的一種工具,需要的朋友可以參考下2023-07-07

