pytorch储存图像saveimages

在训练或者别的过程中储存图像的方法

此处省略数据转换和载入dataset的步骤

from torchvision.utils import save_imageimg_num=0
for _ in range(10):for img,label in dataset:save_image(img,'img'+str(img_num)+'.png')img_num+=1