Pytorch自定义图片数据集
本篇博客旨在实现pytorch读取图片并自定义图片数据集
图像加载方法
主流的图像加载方法主要有三种
下表中xxx表示图片的路径
库
函数/方法
返回值
图像像素格式
像素值范围
图像矩阵表示
skimage
io.imread(xxx)
numpy.ndarray
RGB
[0, 255]
(H X W X C)
cv2
cv2.imread(xxx)
numpy.ndarray
BGR
[0, 255]
(H X W X C)
Pillow(PIL)
Image.open(xxx
原创
2021-10-04 21:09:59 ·
1216 阅读 ·
0 评论