PyTorch载入图片ToTensor,PIL和OpenCV读取图片plt.imread和PIL.Image.open

PyTorch在做一般的深度学习图像处理任务时,先使用dataset类和dataloader类读入图片,在读入的时候需要做transform变换,其中transform一般都需要ToTensor()操作,将dataset类中__getitem__()方法内读入的PIL或CV的图像数据转换为torch.FloatTensor。另外,PIL图像在转换为numpy.ndarray后,格式为(h,w,c),像素顺序为RGB
OpenCV在cv2.imread()后数据类型为numpy.ndarray,格式为(h,w,c),像素顺序为BGR

https://www.cnblogs.com/ocean1100/p/9494640.html

plt.imread和PIL.Image.open读入的都是RGB顺序,而opencv中cv2.imread读入的是BGR通道顺序 。cv2.imread会显示图片更蓝一些。

cv2.imread 和PIL中的image.open图片显示区别 格式互相转换

https://blog.csdn.net/enter89/article/details/90262569

读取图片预测分类

https://www.cnblogs.com/xiximayou/p/13390166.html

你可能感兴趣的:(PyTorch载入图片ToTensor,PIL和OpenCV读取图片plt.imread和PIL.Image.open)