之前遇到PIL、CV2、pyTorch格式读入的图片,总是傻傻分不清,现在特地总结记录一下,希望帮助到有需求的人,同时担心自己忘记。
0 参考链接
1 使用PIL读取图片
总结:PIL读入图片,通道默认为RGB顺序,读出的变量img类型为JPEG类型,size为(width,height),但是为彩色三通道图像。
# _*_ coding:utf-8 _*_
import torch
import torchvision
import torch.utils.data as Data
import cv2
from PIL import Image
import numpy as np
if __name__ == '__main__':
# 使用PIL读取图片
src_pic_path = 'C:\\Users\\pfm\\Desktop\\sz.jpg'
img = Image.open(src_pic_path)
#PIL图片格式
print(img.format) # 'JPEG'
# PIL图片的大小
print(img.size) # (width,height)
# PIL图片通道顺序
print(img.mode) # RGB
2 使用cv2读取图片
总结:cv2读入图片,通道默认顺序是BGR,读出的变量类型为numpy.ndarray,如果为彩色图读入,则尺寸为(height, weight, channel)