读取图像
读取图像我一般采用的有两种方法,如果想要读成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
image = cv2.imread('D:/Graph/1.png', cv2.IMREAD_GRAYSCALE)
cv2.imwrite(path, image)
plt.imshow(image, cmap='gray')
plt.show()
plt.savefig(path)
image = Image.open('D:/Graph/1.png')
总结
- 读取图片如果需要的是
numpy
格式,那么就用opencv
读取;如果需要的是Image格式
的对象,方便使用pytorch
自带的transform
包,就用PIL包
来读。 - 显示图片用
plt
包 - 保存图片用
cv2.imwrite(path, name)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)