python学习(1)图像数据读取方式

图像数据读取方式

一、OpenCV读取图片

OpenCV读取的图片，直接就是numpy.ndarray格式，无需转换

import cv2
dirpath = './image/1.jpg'
img_cv = cv2.imread(dirpath)
print('img_cv:',img_cv.shape)
print('img_cv:',type(img_cv))
print(img_cv)

#看下读取的数据怎么样

img_cv: (1075, 1919, 3)
img_cv: <class 'numpy.ndarray'>
[[[ 88 106 137]
  [ 88 106 137]
  [ 88 106 137]
  ...
  [131  91 109]
  [134  94 112]
  [136  96 114]]
  ...

二、PIL读取图片

PIL读取的图片并不是直接的numpy.ndarray格式，需要进行转换

dirpath = './image/1.jpg'
from PIL import Image
import numpy as np
img_pil = Image.open(dirpath)  #读取的非numpy.ndarray格式
print('img_pil1:',img_pil)
print('img_pil1:',type(img_pil))
img_pil = np.array(img_pil)    #格式转换
print('img_pil2:',img_pil.shape)
print('img_pil2:',type(img_pil))
print('img_pil2:',img_pil)

运行结果

 img_pil1: <PIL.JpegImagePlugin.JpegImageFile image mode=RGB size=1919x1075 at 0x1AA2337B490>
img_pil1: <class 'PIL.JpegImagePlugin.JpegImageFile'>
img_pil2: (1075, 1919, 3)
img_pil2: <class 'numpy.ndarray'>
img_pil2: [[[137 106  88]
  [137 106  88]
  [137 106  88]
  ...
  [109  91 131]
  [112  94 134]
  [114  96 136]]

 [[144 113  95]
  [144 113  95]
  [144 113  95]
  ...

三、keras读取图片

keras深度学习的框架，里面也是内置了读取图片的模块，该模块读取的也不是数组格式，需要进行转换。(未装keras因此结果为他人跑的)

from keras.preprocessing.image import array_to_img, img_to_array 
load_imgimg_keras = load_img(dirpath)#读取数据
  
print("img_keras:",img_keras)
img_keras: <PIL.JpegImagePlugin.JpegImageFile image mode=RGB size=2736x1856 at 0x2201D184BC8>

print("img_keras:",type(img_keras))
img_keras: <class 'PIL.JpegImagePlugin.JpegImageFile'>
 
#使用keras里的img_to_array()
img_keras = img_to_array(img_keras)
print("img_keras:",img_keras.shape)
img_keras: (1856, 2736, 3)
print("img_keras:",type(img_keras))
img_keras: <class 'numpy.ndarray'>
 
 
#可以使用使用np.array()进行转换
mg_keras= np.array(img_keras)

四、skimage读取图片

scikit-image是基于scipy的一款图像处理包，它将图片作为numpy数组进行处理，读取的数据正好是numpy.ndarray格式。

import skimage.io as io
img_io = io.imread(dirpath)
print('img_io:',img_io.shape)
print('img_io:',type(img_io))
print(img_io)

img_io : <class 'numpy.ndarray'
img_io: (1075, 1919, 3)
img_io: <class 'numpy.ndarray'>
[[[137 106  88]
  [137 106  88]
  [137 106  88]
  ...
  [109  91 131]
  [112  94 134]
  [114  96 136]]

五、matplotlib.image读取图片

利用matplotlib.image读取的图片，直接就生成了数组格式

import matplotlib.image as mpig
img_mpig = mpig.imread(dirpath)
print('img_mpig:',img_mpig.shape)
print('img_mpig:',type(img_mpig))

img_mpig: (1075, 1919, 3)
img_mpig: <class 'numpy.ndarray'>

六、matplotlib.pyplot读取图片

利用matplotlib.pyplot读取的图片，同样也是直接就生成了数组格式

import matplotlib.pyplot as plt
img_plt = plt.imread(dirpath)
print('img_plt:',img_plt.shape)
print('img_plt:',type(img_plt))

img_plt: (1075, 1919, 3)
img_plt: <class 'numpy.ndarray'>