【转载】 opencv, PIL.Image的彩色图片维度 && caffe和pytorch的矩阵维度

原文地址：

https://blog.csdn.net/u011668104/article/details/82718375

---------------------------------------------------------------------------------------------

opencv读取彩色图片：BGR，(h,w,c)

例如：

cv2.imread(‘test.jpg’)

Skimage读彩色图片：RGB，(h,w,c)

例如：

from skimage import io

io.imread(‘test.jpg’,as_grey=False)

PIL.Image读取彩色图片：RGB， size:(w,h)，转成numpy后变成(h,w,c)

例如：

image=Image.open(‘test.jpg’)

image = np.array(image,dtype=np.float32)

pytorch tensor matrix: (N, C, H, W)

caffe中imagedata layer彩色图片：BGR

caffe matrix: (N, C, H, W)

参考文献：OpenCV、Skimage、PIL图像处理的细节差异

常用：

from PIL import Image

Image.open(img_path).convert(‘RGB’)

注：有时候用别人的模型需要根据作者对彩色图三个通道的顺序要求

例如，VGG Face 2 中要求“Please note that the input mean vector is in BGR order.”，用它的模型做初始化就只能用BGR顺序。