python读取、显示、保存图片的几种方法

你的名字 2024-05-08 06:48 7阅读 0赞

#### 目录 ####

*  读取图像
 *   *  使用opencv读取
     *  使用PIL包读取
 *  保存图片
 *  显示图片
 *  代码
 *  总结

## 读取图像 ##

读取图像我一般采用的有两种方法，如果想要读成`numpy数组`格式就用`opencv`方式，想要读成`Image对象`格式就用`PIL包`读取。

### 使用opencv读取 ###

如果是读取灰度图，需要指定`颜色模式`为`cv2.IMREAD_GRAYSCALE`，这样读取出的是一个`二维数组`，而不是彩色图像的`三维数组`。

如果读取彩色图像，则不需要指定读取模式，这样读取出的是一个三维数组(H, W, C)，但通道的的顺序是`BGR`。

### 使用PIL包读取 ###

使用PIL包读取图片时不需要指定`颜色模式`，程序会自动判断。

使用这个包的好处在于会读出来一个Image对象，方便后续`pytorch`的`transform`方法使用。

## 保存图片 ##

保存图片就用`cv2.imwrite(path, image)`就够了。

## 显示图片 ##

显示图片用`plt`包。

## 代码 ##

import cv2
    import matplotlib.pyplot as plt
    from PIL import Image
    import numpy as np
    
    # 使用opencv读取图片
    image = cv2.imread('D:/Graph/1.png', cv2.IMREAD_GRAYSCALE)
    cv2.imwrite(path, image)
    # plt.show函数需要指明颜色模式
    plt.imshow(image, cmap='gray')
    plt.show()
    plt.savefig(path)
    
    # 方法二：使用PIL包读取
    # 这种方式不需要指定图像的模式，程序会自动识别：灰度图像的维度为2，彩色图像的维度为3
    # 读取出来的对象不是numpy数组，而是一个Image对象，使用这种方式是为了方便使用pytorch官方的transform模块
    image = Image.open('D:/Graph/1.png')

## 总结 ##

*  读取图片如果需要的是`numpy`格式，那么就用`opencv`读取；如果需要的是`Image格式`的对象，方便使用`pytorch`自带的`transform`包，就用`PIL包`来读。
 *  显示图片用`plt`包
 *  保存图片用`cv2.imwrite(path, name)`