1.图像读取
用pillow读取,该库较为常用,且比较简单。
调用下面语句即可:im =Image.open(cat.jpg’)
2.数据扩增
要使训练结果更好,更具有泛化性能,需要对数据进行扩增。
torchvison库中自带的的方法有如下图
此外,还可调用其他库或自己编写。
例如im2 = im.filter(ImageFilter.BLUR)进行图片模糊操作。
3.Pytorch读取图像
首先需要将所给训练集变成Pytorch可读取的dataset格式,代码如下
然后用train_loader载入即可