PIL读取图片
import os
import numpy as np
from PIL import Image
from matplotlib import pyplot as plt
def load_image(image_name,data_dir,image_extension= '.jpg'):
'''
PIL或者plt读取图片是以RGB的格式显示
'''
path = os.path.join(data_dir,image_name + image_extension)
image_PIL = np.asarray(Image.open(path).convert('RGB'))
image_cv2 = cv2.imread(path) #读取的直接是BGR
return image_PIL
data_dir = 'C:/Users/Administrator/Desktop/python_learning/'
image_PIL= load_image('000001',data_dir)
plt.imshow(image_PIL)
显示结果:
将图片转换成GBR
image_PIL_GBR = image_PIL[:,:,::-1].copy()
plt.imshow(image_PIL_GBR)
显示结果:
cv2读取图片
import os
import numpy as np
from PIL import Image
from matplotlib import pyplot as plt
import cv2
def load_image(image_name,data_dir,image_extension= '.jpg'):
'''
cv2读取图片是以BGR的格式显示
'''
path = os.path.join(data_dir,image_name + image_extension)
image_cv2 = cv2.imread(path) #读取的直接是BGR
return image_cv2
data_dir = 'C:/Users/Administrator/Desktop/python_learning/'
image_cv2 = load_image('000001',data_dir)
plt.imshow(image_cv2)
显示结果: