把通道放在前面:
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不能解决通道转换问题
pycaffe做识别时通道转换问题:
要注意一点的就是:Caffe中彩色图像的通道是BGR格式,图像存储是【0,255】
1.caffe.io.load_image方式 view plai cop
image = caffe.io.load_image(image_file) #加载图片
使用caffe.io.load_image()读进来的是RGB格式和0~1(float)
所以在进行识别之前要在transformer中设置transformer.set_raw_scale('data',255)(缩放至0~255)
以及transformer.set_channel_swap('data',(2,1,0)(将RGB变换到BGR)
# python中将图片存储为[0, 1],而caffe中将图片存储为[0, 255],所以需要一个转换 transformer.set_raw_scale('data', 255) # 缩放到[0,255]之间 transformer.set_channel_swap('data', (2,1,0)) #交换通道,将图片由RGB变为BGR(caffe中图片是BGR格式,而原始格式是RGB,所以要转化)
2.使用cv2.imread()来读取图片
cv2.imread()接口读图像,读进来直接是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来读取图片
对于彩色图像,不管其图像格式是PNG,还是BMP,或者JPG,在PIL中,使用Image模块的open()函数打开后,返回的图像对象的模式都是“RGB”。而对于灰度图像,不管其图像格式是PNG,还是BMP,或者JPG,打开后,其模式为“L”。所以需要转换格式,但不需要缩放到[0,255]
data = np.array(Image.open(self.dataRoot+img_list)) data = np.transpose(data,(2,0,1))#转换通道 data[[0,2],...] = data[[2,0],...] #RGB→BGR
4.对于matlab来说
Caffe中的blobs格式是N*C*H*W,分别是数量Number,通道数Channel,以及宽度Height和宽度Width
而matlab中是先宽后高,即[w,h],图像的通道是RGB
所以需要进行相应的转换:
im_data = im (:,:,[3,2,1]) ; %RGB to BGR
im_data = permute(im_data, [2,1,3]); %旋转高度和宽度
最后,分享一个Caffe的典型python识别代码:
# -*- 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 #训练的图片尺寸 image_file = 'F:/.../.../nihao.jpg'#图片路径 model_def = 'F:/.../.../deploy.prototxt'#deploy模型文件位置 model_weights = 'F:/.../.../_iter_20000.caffemodel'#训练完的模型位置 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) ############################################################################## #图片预处理 transformer = caffe.io.Transformer({'data': net.blobs['data'].data.shape}) ##设定图片的shape格式(1,3,227,227),大小由deploy 文件指定 # python读取的图片文件格式为H×W×K,需转化为K×H×W transformer.set_transpose('data', (2,0,1)) #改变维度的顺序,由原始图片(227,227,3)变为(3,227,227) transformer.set_mean('data', mu) # 每个通道减去均值 # python中将图片存储为[0, 1],而caffe中将图片存储为[0, 255],所以需要一个转换 transformer.set_raw_scale('data', 255) # 缩放到【0,255】之间 transformer.set_channel_swap('data', (2,1,0)) #交换通道,将图片由RGB变为BGR(caffe中图片是BGR格式,而原始格式是RGB,所以要转化) net.blobs['data'].reshape(1,3,size, size) # 将输入图片格式转化为合适格式(与deploy文件相同) #上面这句,第一参数:图片数量 第二个参数 :通道数 第三个参数:图片高度 第四个参数:图片宽度 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() # 给出概率最高的是第几类,需要自己对应到我们约定的类别去
以上这篇python与caffe改变通道顺序的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持小牛知识库。
问题内容: Goroutines在通道上的阻塞顺序是否决定了它们将解除阻塞的顺序?我不关心所发送消息的顺序(它们一定会被排序),但是不会影响Goroutines的顺序。 想象一下在多个Goroutine(1、2和3)之间共享一个空通道,每个Goroutine都尝试在上接收消息。由于为空,每个Goroutine将阻塞。当我向发送消息时,Goroutine 1将首先解锁吗?还是2或3可能会收到第一条消
问题内容: 我有一个包含JSON对象的文件。已通过以下方式加载: 此时,input_data仅包含一个字符串,现在我将其解析为JSON: data_content具有我需要的字符串的JSON表示形式,但是由于某种原因,在json.loads之后我不清楚,它改变了键的原始顺序,例如,如果我的文件包含以下内容: 在json.loads之后,顺序更改为类似以下内容: 为什么会这样呢?有办法保留订单吗?我
本文向大家介绍python方法解析顺序?相关面试题,主要包含被问及python方法解析顺序?时的应答技巧和注意事项,需要的朋友参考一下 参考回答: Python的方法解析顺序优先级从高到低为:实例本身类继承类(继承关系越近,越先定义,优先级越高)
问题内容: 我有以下: 我通过分配添加了更多列: 如何将列mean移到最前面,即将其设置为第一列,而其他列的顺序保持不变? 问题答案: 你还可以执行以下操作: 你可以通过以下方式获取列列表: 输出将产生: …然后轻松将其放到第一个功能中即可手动重新排列
问题内容: 这个问题已经在这里有了答案 : Pyplot自动对y值排序 (2个答案) 3年前关闭。 在此示例中: 图中的条按字母顺序排序。 如何使它们以与代码中相同的顺序显示(从最高到最低)? 问题答案: Matplotlib 2.1提供了新选项,可将字符串用作其绘图功能的输入-称为“类别”。这是一项新功能,尚未完全正常运行。一个警告是字符串会自动(按字母顺序)排序。此限制将在下一版本中删除。 在
我试图从一个带有Xpath的网页中提取文本,如下所示 使用python代码 这最终会产生以下错误: NoSuchElementException:没有这样的元素:无法找到元素:{“方法”:“xpath”,“选择器”:“/*[@id=“twocols”]/tbody/tr/td[2]/table/tbody/tr[7]/td[2]”} xpath末尾的节点tr[7]变为tr[6],因为有时会在页面中