Python读取保存图像文件

Python处理图像数据时通常需要对图像文件进行读取、保存等操作，因此将现有的方法归纳了一下。

1. PIL

依赖包：Pillow

安装：pip install Pillow

源码：

import numpy as np
from PIL import Image
import matplotlib.pyplot as plt

image = Image.open("image.jpg")

image_resize = image.resize((128, 128))
image_array = np.array(image_resize)
image_output = Image.fromarray(image_array)
image_output.save("image_output.jpg")

plt.imshow(image_output)
plt.axis("off")
plt.show()

2. opencv

依赖包：opencv-python

安装：pip install opencv-python

源码：

import cv2

image = cv2.imread("image.jpg")

image_gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
image_resize = cv2.resize(image, (128, 128))

cv2.imwrite("image_output.jpg", image_gray)
cv2.namedWindow("img_out")
cv2.imshow('img_out',image_gray)
cv2.waitKey(0)
cv2.destroyAllWindows()

3. imageio

依赖包：imageio

安装：pip install imageio

源码：

from imageio import imread, imsave
from scipy import ndimage
import matplotlib.pyplot as plt

image = imread("image.jpg")

img_rote_20 = ndimage.rotate(image, 20)
imsave("image_out.jpg", img_rote_20)

plt.imshow(img_rote_20)
plt.axis("off")
plt.show()

4. skimage (recommend to use)

依赖包：scikit-image

安装：pip install scikit-image

源码：

from skimage import io
import matplotlib.pyplot as plt

image = io.imread("image.jpg")

io.imsave("image_output.jpg", image)

io.imshow(image)
plt.show()

5. matplotlib

依赖包：matplotlib

安装：pip install matplotlib

源码：

import matplotlib.image as mpimg
import matplotlib.pyplot as plt

image = mpimg.imread("image.jpg")

mpimg.imsave("image_output.jpg", image)

plt.imshow(image)
plt.axis("off")
plt.show()

7. keras (machine learning or deep learning)

依赖包：keras

安装：pip install keras

源码：

from keras.preprocessing import image

img = image.load_img("image.jpg", target_size=(128, 128))

image_array = image.img_to_array(img)
image_output = image.array_to_img(img)

image_output.save("image_output.jpg")

 

上述七种方法均可以对图像文件进行读取保存处理，对日常cv预处理图像非常有帮助。